update

app.py

@@ -53,7 +53,7 @@ def get_description(task_name):
     task = task_names_to_tasks[task_name]
     return task_descriptions[task]
 
-#@spaces.GPU(duration=10)
+@spaces.GPU(duration=60)
 def predict_single_label(smiles, task_name):
     task = task_names_to_tasks[task_name]
 

utils.py

@@ -209,7 +209,6 @@ class ReactionPredictionModel():
         self.forward_model.to("cuda")
         self.forward_model.eval()
 
-    @spaces.GPU(duration=60)
     def predict(self, test_loader, task_type):
         predictions = []
         for i, batch in tqdm(enumerate(test_loader), total=len(test_loader), desc="Evaluating"):
@@ -219,8 +218,6 @@ class ReactionPredictionModel():
             del inputs['token_type_ids']
 
             if task_type == "retrosynthesis":
-                self.retro_model.to("cuda")
-                self.retro_model.eval()
                 inputs = {k: v.to(self.retro_model.device) for k, v in inputs.items()}
                 print(inputs)
                 print(self.retro_model.device)
@@ -233,8 +230,6 @@ class ReactionPredictionModel():
                                    length_penalty=0.0,
                                    )
             else:
-                self.forward_model.to("cuda")
-                self.forward_model.eval()
                 inputs = {k: v.to(self.forward_model.device) for k, v in inputs.items()}
                 print(inputs)
                 print(self.forward_model.device)