Update app.py

app.py

@@ -6,16 +6,17 @@ subprocess.run(
     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
     shell=True,
 )
-subprocess.run(
-    "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,
-)
+# subprocess.run(
+#     "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__)
+print(torch.cuda.is_available())
 from huggingface_hub import snapshot_download 
 snapshot_download(repo_id="jzq11111/mooncast", local_dir='./resources/')