Update app.py

app.py

@@ -35,7 +35,6 @@ models = {
     'Cellular Component': get_model('Cellar Component'),
     }
 
-
 # Load the mistral model
 mistral_model = MistralForCausalLM.from_pretrained("teknium/OpenHermes-2.5-Mistral-7B", torch_dtype=torch.float16)
 mistral_model.to('cuda')
@@ -183,11 +182,11 @@ def generate_caption(protein, prompt):
         return res_str
     res_str = ''
     if len(union_pred_terms[0]) != 0:
-        res_str += f"Based on the given amino acid sequence, the protein appears to have a primary function of {', '.join(pred_terms)}. "
+        res_str += f"Based on the given amino acid sequence, the protein appears to have a primary function of {', '.join(union_pred_terms[0])}. "
     if len(union_pred_terms[1]) != 0:
-        res_str += f"It is likely involved in the {', '.join(pred_terms)}. "
+        res_str += f"It is likely involved in the {', '.join(union_pred_terms[1])}. "
     if len(union_pred_terms[2]) != 0:
-        res_str += f"It's subcellular localization is within the {', '.join(pred_terms)}."
+        res_str += f"It's subcellular localization is within the {', '.join(union_pred_terms[2])}."
     return res_str
     # return "test"