Update app.py

app.py

@@ -14,6 +14,18 @@ import csv
 from datetime import datetime
 import json
 
+
+zero = torch.tensor([0]).cuda()
+one = torch.tensor([0]).cuda()
+two = torch.tensor([0]).cuda()
+
+@spaces.GPU
+def test(n):
+    print('in', zero.device, one.device, two.device)
+    return 0
+
+test(1)
+
 from rdkit import Chem
 from rdkit.Chem import Draw