Update app.py

app.py

@@ -6,10 +6,8 @@ zero = torch.Tensor([0]).cuda()
 print(f"Outside function: {zero.device}")  # Should print 'cuda:0'
 
 # Define the function to greet and perform a tensor operation on the GPU
+@spaces.GPU
 def greet(n):
-    # Make sure the tensor is on the same device
-    zero_gpu = zero.to('cuda')  # This ensures the tensor is on the GPU
-    print(f"Inside function: {zero_gpu.device}")  # Should print 'cuda:0'
     result = zero_gpu + torch.Tensor([n]).cuda()  # Perform operation on GPU
     return f"Hello, result: {result.item()}"