Update app.py

app.py

@@ -12,6 +12,8 @@ subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={
 os.system("pip install transformers==4.51.1")
 os.system("pip install numpy==1.26.4")
 os.system("pip install vllm==0.8.2")
+os.system("pip install torch==2.4.0 torchvision==0.18.0 --index-url https://download.pytorch.org/whl/cu124")
+
 import sys
 import threading
 import re