prompt model change

app.py

@@ -52,6 +52,18 @@ pipe = pipeline(
     max_new_tokens=200
 )
 
+def clean_chat_output(full_text):
+    # Split at the last assistant tag (TinyLlama uses <|assistant|>)
+    if "<|assistant|>" in full_text:
+        full_text = full_text.split("<|assistant|>")[-1]
+
+    # Remove any leftover "Item:" echoes
+    lines = full_text.strip().split("\n")
+    if lines[0].lower().startswith("item:"):
+        lines = lines[1:]
+
+    return "\n".join(lines).strip()
+
 
 def explain_recycling(label):
     system_msg = {
@@ -85,7 +97,8 @@ def explain_recycling(label):
         do_sample=False   # deterministic to avoid repeating instructions
     )
 
-    return outputs[0]["generated_text"]
+    raw = outputs[0]["generated_text"]
+    return clean_chat_output(raw)
 
 
 # ------------------ PIPELINE ------------------