Update app.py

app.py

@@ -11,6 +11,11 @@ from PIL import Image, ImageOps
 from torchvision import transforms
 
 
+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
+
 # ----------------- HELPER FUNCTIONS ----------------- #
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
 ASSETS_DIR = os.path.join(os.path.dirname(__file__), "assets")