source in if

ipitio · Oct 9, 2024 · 6c9263a · 6c9263a
1 parent 18cb940
commit 6c9263a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/predict.sh b/src/predict.sh
@@ -4,11 +4,11 @@
 set -Eeuo pipefail
 declare -r pdf_dir="${1:-.}"
 export OMP_THREAD_LIMIT=1
-
 python3 -m venv venv
-source venv/bin/activate
 
 if [[ -f venv/bin/pip3 ]]; then
+    source venv/bin/activate
+
     ./venv/bin/pip3 install -r requirements.txt
     ./venv/bin/python3 ./main.py "$pdf_dir"
 
@@ -17,4 +17,5 @@ if [[ -f venv/bin/pip3 ]]; then
 elif [[ -f /.dockerenv ]]; then
     pip3 install -r requirements.txt --user --break-system-packages
     python3 ./main.py "$pdf_dir"
+    rm -rf venv
 fi