Upload app.py

app.py

@@ -333,9 +333,9 @@ def get_multiple_recommendations(pred_caption: str, llm_model, tokenizer_model,
     final_recommendations = (
         generated_text.replace(getattr(tokenizer_model, "eos_token", ""), "").replace("<|im_end|>", "").strip()
     )
-    final_recommendations = final_recommendations.replace("Cause:", "**Cause**:")
-    final_recommendations = final_recommendations.replace("Immediate Action:", "**Immediate Action**:")
-    final_recommendations = final_recommendations.replace("Long-term Action:", "**Long-term Action**:")
+    final_recommendations = final_recommendations.replace("Cause:", "Cause:")
+    final_recommendations = final_recommendations.replace("Immediate Action:", "Immediate Action:")
+    final_recommendations = final_recommendations.replace("Long-term Action:", "Long-term Action:")
 
     return final_recommendations, retrieve_knowledge(pred_caption, knowledge_base)