Update app.py

app.py

@@ -5,11 +5,8 @@ from huggingface_hub import login
 import os
 import gradio as gr
 
-# 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"
+device = zerogpu.select_device()  # ZeroGPU will automatically choose 'cuda' or 'cpu'
 
 # Check and print if the selected device is GPU or CPU
 if device == "cuda":