Stripped assistant

app.py

@@ -100,11 +100,14 @@ def chat_llama3_8b(message: str,
     t.start()
 
     outputs = []
+    first_chunk = True
     for text in streamer:
+        if first_chunk and text.startswith("assistant"):
+            text = text[len("assistant"):].lstrip(": \n")  # Remove "assistant" and any following symbols
+        first_chunk = False
+
         outputs.append(text)
-        #print(outputs)
-        yield "".join(outputs)
-        
+        yield "".join(outputs)        
 
 # Gradio block
 chatbot=gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Gradio ChatInterface')