Model prompt change

app.py

@@ -1,9 +1,3 @@
-import asyncio
-try:
-    asyncio.get_event_loop()
-except RuntimeError:
-    asyncio.set_event_loop(asyncio.new_event_loop())
-
 import gradio as gr
 import torch
 from transformers import (
@@ -12,7 +6,6 @@ from transformers import (
     T5ForConditionalGeneration,
     pipeline
 )
-
 # ------------------ LOAD CLASSIFIER ------------------
 cls_model_name = "yangy50/garbage-classification"
 classifier = pipeline("image-classification", model=cls_model_name)
@@ -44,6 +37,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(class_label):
     system_msg = {
         "role": "system",
@@ -74,7 +79,8 @@ def explain_recycling(class_label):
         top_p=0.9
     )
 
-    return outputs[0]["generated_text"]
+    raw = outputs[0]["generated_text"]
+    return clean_chat_output(raw)
 
 
 # ------------------ PIPELINE ------------------