Update app.py

app.py

@@ -1,7 +1,20 @@
 import gradio as gr
+import torch
+import spaces  # Import ZeroGPU helper
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Function to check GPU availability
+@spaces.GPU  # This tells Hugging Face to allocate a GPU when this function is called
+def check_gpu():
+    if torch.cuda.is_available():
+        return "✅ GPU is available! CUDA device: " + torch.cuda.get_device_name(0)
+    else:
+        return "❌ No GPU detected!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch(share=True, debug=True)
+# Gradio UI
+iface = gr.Interface(
+    fn=check_gpu,  # Call the GPU-checking function
+    inputs=[],
+    outputs="text",
+)
+
+iface.launch(share=True, ssr=False)