Update app.py

app.py

@@ -5,11 +5,14 @@ from huggingface_hub import login
 import os
 import gradio as gr
 
-# Automatically select the available device (GPU if available, otherwise CPU)
-device = zerogpu.select_device()  # This will automatically choose 'cuda' or 'cpu'
+# Initialize ZeroGPU, which will manage device selection
+zerogpu.init()  # Initializes ZeroGPU
+
+# Automatically choose GPU if available, otherwise CPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Check and print if the selected device is GPU or CPU
-if torch.cuda.is_available():
+if device == "cuda":
     print(f"Using GPU: {torch.cuda.get_device_name()}")
 else:
     print("Using CPU")
@@ -72,4 +75,3 @@ iface = gr.Interface(
     outputs="image"
 )
 iface.launch()
-