Update app.py

app.py

@@ -82,11 +82,14 @@ def save_interesting_log(smiles, properties, suggested_properties):
         }
         writer.writerow(log_data)
 
-print(torch.Tensor([0]).cuda())
+zero = torch.Tensor([0]).cuda()
+print(zero.device)
 
 @spaces.GPU
 def test():
-    print('intest',torch.Tensor([0]).cuda())
+    print('intest', zero.device)
+    device = zero.device
+    print('1 ',torch.Tensor([1]).to(device))
     return 0
 
 test()