Update app.py

app.py

@@ -6,11 +6,16 @@ subprocess.run(
     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
     shell=True,
 )
-
 subprocess.run(
-    "pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121",
+    "pip uninstall torch torchvision torchaudio -y",
+    shell=True,
+)
+subprocess.run(
+    "pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121 --no-cache-dir --force-reinstall",
     shell=True,
 )
+import torch
+print(torch.__version__)
 from huggingface_hub import snapshot_download 
 snapshot_download(repo_id="jzq11111/mooncast", local_dir='./resources/')