Update app.py

app.py

@@ -9,7 +9,11 @@ import os
 os.environ["SAFETENSORS_FAST_GPU"] = "1"
 
 import subprocess
-subprocess.run(['sh', './torch.sh'])
+
+result = subprocess.run(['nvidia-smi'], capture_output=True, text=True, check=True)  # check=True raises an exception if the command fails
+print(result)
+
+#subprocess.run(['sh', './torch.sh'])
 
 #import sys
 #conda_prefix = os.path.expanduser("~/miniconda3")