Update app.py

app.py

@@ -2,21 +2,12 @@ import gradio as gr
 import spaces
 import torch
 
-def gpu_info():
-    info = []
-    info.append(f"Torch Version: {torch.__version__}")
-    info.append(f"CUDA Available: {torch.cuda.is_available()}")
-    info.append(f"GPU Count: {torch.cuda.device_count()}")
-    
-    if torch.cuda.is_available():
-        info.append(f"GPU Name: {torch.cuda.get_device_name(0)}")
-        info.append(f"VRAM: {torch.cuda.get_device_properties(0).total_memory // (1024**2)} MB")
-    else:
-        info.append("No GPU detected.")
-    
-    return "\n".join(info)
+@spaces.GPU
+def greet(n):
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    zero = torch.tensor([0], device=device)
+    print(zero.device)  # This should print 'cuda:0' if GPU is available
+    return f"Hello {zero + n} Tensor"
 
-# Gradio Interface
-iface = gr.Interface(fn=gpu_info, inputs=None, outputs="text", title="GPU Information Checker")
-
-iface.launch()
+demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
+demo.launch()