Update app.py

app.py

@@ -3,11 +3,15 @@ import spaces
 import torch
 
 zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
+print('zero device: ' + zero.device)
+
+one = torch.ones(2, 2, device='cuda')
+print('one device: ' + one.device)
 
 @spaces.GPU
 def greet(n):
-    print(zero.device) # <-- 'cuda:0' 🤗
+    print('zero device' + zero.device)
+    print('one device' + zero.device)
     return f"Hello {zero + n} Tensor"
 
 gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text()).launch()