Update app.py

app.py

@@ -6,14 +6,11 @@ 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(
+    "nvidia-smi",
+    shell=True,
+)
+
 import torch
 print(torch.__version__)
 print(torch.cuda.is_available())