Update app.py

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 import spaces
 import torch
 
-zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
-
 @spaces.GPU
 def greet(n):
-    print(zero.device) # <-- 'cuda:0' 🤗
+    # Create the tensor inside the GPU worker where a real GPU exists
+    zero = torch.Tensor([0]).to("cuda")
+    print(zero.device)  # This will print 'cuda:0'
     return f"Hello {zero + n} Tensor"
 
 demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
-demo.launch()
+demo.launch()