Update app.py

app.py

@@ -30,6 +30,11 @@ client_image = Client("measmonysuon/DALLE-4K")
 user_info = whoami(token=API_TOKEN)
 print("User Info:", user_info)
 
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla
+      
 # Define resolutions and default style
 resolutions = {
     "896x1152": (896, 1152),