Update app.py

app.py

@@ -4,8 +4,8 @@ import os
 import logging
 import requests
 from huggingface_hub import login, whoami, HfFolder, HfApi, SpaceHardware
-import torch
 import time
+import torch
 
 webhook_server = os.getenv('webhook_server')
 API_TOKEN = os.getenv('ZeroGPU')
@@ -30,12 +30,12 @@ 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")
+#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-print(torch.version.cuda)
-print(f"Is CUDA available: {torch.cuda.is_available()}")
+#print(torch.version.cuda)
+#print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+#print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 # Tesla
       
 # Define resolutions and default style