Update app.py

app.py

@@ -83,13 +83,13 @@ def save_interesting_log(smiles, properties, suggested_properties):
         writer.writerow(log_data)
 
 zero = torch.Tensor([0]).cuda()
-print(zero.device)
+print('outside',zero.device)
 
 @spaces.GPU
 def test():
     print('intest device', zero.device)
     device = zero.device
-    print('1 ',torch.Tensor([1]).cuda())
+    # print('1 ',torch.Tensor([1]).cuda())
     return 0
 
 test()