Update app.py

app.py

@@ -218,8 +218,7 @@ def predict(
         do_sample=True
     )[0]["generated_text"]
 
-    formatted_output += output.strip() + solution_prefix
-    formatted_output += "Final Answer (This part is a placeholder and needs better extraction): ... "
+    formatted_output += output.strip()
 
     return formatted_output
 
@@ -252,8 +251,7 @@ def compare_models(
         thinking_prefix = f"**{model_name} - Thinking Process:**\n"
         solution_prefix = f"\n**{model_name} - Solution:**\n"
         formatted_output = thinking_prefix
-        formatted_output += raw_output.strip() + solution_prefix
-        formatted_output += f"{model_name} Final Answer: ... "
+        formatted_output += raw_output.strip()
         return formatted_output
 
     local_out_raw = local_pipe(
@@ -413,8 +411,7 @@ def chat_rag(
         do_sample=True
     )[0]["generated_text"]
 
-    formatted_output += output.strip() + solution_prefix
-    formatted_output += "Final Answer (This part is a placeholder and needs better extraction): ... "
+    formatted_output += output.strip()
     assistant_reply = formatted_output
 
     if assistant_reply.startswith(prompt):