Update app.py

app.py

@@ -23,10 +23,10 @@ async def chat_endpoint(message: str = Form(...)):
         llm_model_cache["llm"] = get_llm()
 
     prediction = classification_modeL_cache["model"].predict(message)
-    sentiment = "Positive" if prediction[0][1] > 0.5 else "Negative"
+
 
     result = llm_model_cache["llm"].invoke(
-        prompt.format(text=message, prediction=sentiment))
+        prompt.format(text=message, positive_prob=prediction[0][1], negative_prob=prediction[0][0]))
 
     if result:
         return result.content