Update app.py

app.py

@@ -23,7 +23,11 @@ print('after spaces', zero.device)
 @spaces.GPU
 def test(n):
     print(n)
-    print('inside device', zero.device)
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    print('intest device', device)
+    print('inside zero old', zero.device)
+    zero = zero.to(device)
+    print('inside zero new', zero.device)
     # device = zero.device
     print('1 ',torch.Tensor([1]).cuda())
     return 0