Update app.py

app.py

@@ -1,18 +1,14 @@
 import gradio as gr
+import spaces
 import torch
 
-# Initialize a tensor on the GPU
-zero = torch.Tensor([0]).cuda() 
-print(f"Outside function: {zero.device}")  # Should print 'cuda:0'
+ zero = torch.Tensor([0]).cuda()
+print(zero.device) # <-- 'cpu' 🤔
 
-# Define the function to greet and perform a tensor operation on the GPU
-@spaces.GPU
+ @spaces.GPU
 def greet(n):
-    result = zero_gpu + torch.Tensor([n]).cuda()  # Perform operation on GPU
-    return f"Hello, result: {result.item()}"
+    print(zero.device) # <-- 'cuda:0' 🤗
+    return f"Hello {zero + n} Tensor"
 
-# Create Gradio interface
-demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
-
-# Launch the Gradio demo
+ demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
 demo.launch()