Update app.py

app.py

@@ -14,18 +14,6 @@ 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
 
@@ -65,6 +53,7 @@ atexit.register(cleanup_temp_files)
 def random_properties():
     return known_labels[all_properties].sample(1).values.tolist()[0]
 
+@spaces.GPU
 def load_model(model_choice):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = load_graph_decoder(path=model_choice)