Update utils.py

utils.py

@@ -207,7 +207,7 @@ class ReactionPredictionModel():
         self.forward_model.config.pad_token_id = self.tokenizer.pad_token_id
         self.forward_model.to("cuda")
 
-    @spaces.GPU(duration=20)
+    @spaces.GPU(duration=30)
     def predict(self, test_loader, task_type):
         predictions = []
         for i, batch in tqdm(enumerate(test_loader), total=len(test_loader), desc="Evaluating"):
@@ -258,7 +258,7 @@ class ReactionPredictionModel():
                 predictions.append(canonized_smiles_list)
         
         rank, invalid_rate = compute_rank(predictions)
-
+        print(predictions, rank)
         return rank
     
     def predict_single_smiles(self, smiles, task_type):