Update app.py

app.py

@@ -30,6 +30,8 @@ client_image = Client("measmonysuon/DALLE-4K")
 user_info = whoami(token=API_TOKEN)
 print("User Info:", user_info)
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 print(torch.version.cuda)
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True