Update app.py

app.py

@@ -154,6 +154,11 @@
 #     demo.launch(share=True)
 
 
+import torch
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
 
 import gradio as gr
 import shutil