Update app.py

app.py

@@ -14,6 +14,7 @@ from contextlib import contextmanager
 from gradio.helpers import create_examples 
 import subprocess
 
+import torch; print(torch.cuda.device_count()) 
 
 from pathlib import Path
 import os