diff --git a/py/venv.sh b/py/venv.sh
index 4a3be2f..ed72fee 100644
--- a/py/venv.sh
+++ b/py/venv.sh
@@ -1,5 +1,7 @@
 #!/bin/bash
 
+rm -rf venv/
 virtualenv venv
 source venv/bin/activate
 pip install transformers
+pip install torch