Update app.py

app.py

@@ -8,10 +8,14 @@ print('zero device: ' + str(zero.device))
 one = torch.ones(2, 2, device='cuda')
 print('one device: ' + str(one.device))
 
+two = torch.matmul(one, one, device='cuda')
+print('two device: ' + str(two.device))
+
 @spaces.GPU
 def greet(n):
-    print('zero device' + str(zero.device))
-    print('one device' + str(one.device))
+    print('zero device: ' + str(zero.device))
+    print('one device: ' + str(one.device))
+    print('two device: ' + str(two.device))
     return f"Hello {zero + n} Tensor"
 
 gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text()).launch()