Update app.py

app.py

@@ -5,7 +5,7 @@ import argparse
 import json
 import os
 
-os.system("pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124")
+
 os.system("pip install matplotlib==3.10.6")
 os.system("pip install numpy==1.26.4")
 os.system("pip install Pillow==11.3.0")
@@ -22,7 +22,7 @@ import subprocess
 subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 import sys
 import threading
-
+os.system("pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124")
 
 import re
 from typing import Any, Dict, List