Update app.py

app.py

@@ -55,14 +55,22 @@ def predict(message, history):
     partial_message = ""
     for new_token in streamer:
         partial_message += new_token
-        if '</s>' in partial_message:  # Breaking the loop if the stop token is generated.
+        if '<|im_end|>' in partial_message:  # Breaking the loop if the stop token is generated.
             break
         yield partial_message
 
 
 # Setting up the Gradio chat interface.
 gr.ChatInterface(predict,
-                 title="Phixtral 2x2.8 Chatbot",
-                 description="Ask Phixtral any questions",
-                 examples=['Write an epic poem about Ancient Rome.', 'Write Python code to print the Fibonacci sequence.']
-                 ).launch()  # Launching the web interface.
+                 title="🔀 Phixtral 2x2_8 Chatbot",
+                 description="![](https://i.imgur.com/UOb2fvh.jpg)\n\nChat with [mlabonne/phixtral-4x2_8](https://huggingface.co/mlabonne/phixtral-4x2_8), the first Mixture of Experts made by merging two fine-tuned [microsoft/phi-2](https://huggingface.co/microsoft/phi-2) models. This small model (4.46B param) is good in various tasks, such as programming, story writing, and more.",
+                 examples=[
+                     'Can you solve the equation 2x + 3 = 11 for x?',
+                     'Write an epic poem about Ancient Rome.',
+                     'Who was the first person to walk on the Moon?',
+                     'Use a list comprehension to create a list of squares for numbers from 1 to 10.',
+                     'Recommend some popular science fiction books.',
+                     'Can you write a short story about a time-traveling detective?'
+                 ],
+                 theme=gr.themes.Soft(),
+                 ).launch()