Update app.py

app.py

@@ -31,8 +31,13 @@ def models(text, model="Mixtral 8x7B"):
     for response in stream:
         if not response.token.text == "</s>":
             output += response.token.text
+                if not response.token.text == "<|assistant|>":
+                    output += response.token.text
 
     return output
 
-demo = gr.Interface(fn=models, inputs=["text", gr.Dropdown([ 'Mixtral 8x7B','Llama 3 8B','Mistral 7B v0.3','Phi 3 mini', ], value="Mistral 7B v0.3", label="Select Model") ], outputs="text", live=True, batch=True, max_batch_size=1000)
+description="""# Chat GO
+### Inspired from Goggle Go"""
+
+demo = gr.Interface(description=description,fn=models, inputs=["text", gr.Dropdown([ 'Mixtral 8x7B','Llama 3 8B','Mistral 7B v0.3','Phi 3 mini', ], value="Mistral 7B v0.3", label="Select Model") ], outputs="text", live=True, batch=True, max_batch_size=1000)
 demo.launch()