Update app.py

app.py

@@ -4,6 +4,14 @@ from huggingface_hub import login
 import os
 import gradio as gr
 
+# Check for GPU availability and set device accordingly
+if torch.cuda.is_available():
+    device = "cuda"
+    print("GPU is available")
+else:
+    device = "cpu"
+    print("GPU is not available, using CPU")
+
 # Retrieve the token from the environment variable
 token = os.getenv("HF_TOKEN")  # Hugging Face token from the secret
 if token: