Update app.py

app.py

@@ -1,5 +1,6 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
+
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def format_prompt(message, history, system_prompt):
@@ -11,9 +12,7 @@ def format_prompt(message, history, system_prompt):
     prompt += f" [SYSTEM] {system_prompt} [/SYSTEM]"
     return prompt
 
-def generate(
-    prompt, history, system_prompt, temperature=0.2, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
-):
+def generate(prompt, history, system_prompt, temperature=0.2, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -34,18 +33,13 @@ def generate(
         yield output
     return output
 
-mychatbot = gr.Chatbot(
-    avatar_images=["./user.png", "./botm.png"],
-    bubble_full_width=False,
-    show_label=False,
-    show_copy_button=True,
-    likeable=True,
+demo = gr.Interface(
+    fn=generate,
+    inputs=[gr.Textbox(lines=2, label="Your Message"), gr.State()],
+    outputs=[gr.Textbox(label="ChatGPT's Response"), gr.State()],
+    title="Tomoniai's Mixtral 8x7b Chat",
+    allow_flagging="never",
+    theme="default"
 )
 
-demo = gr.Interface(fn=generate, 
-                    inputs=[gr.Textbox(lines=2, label="Your Message"), gr.State(), gr.State()],
-                    outputs=[gr.Textbox(label="ChatGPT's Response"), gr.State(), gr.State()],
-                    title="Tomoniai's Mixtral 8x7b Chat",
-                    allow_flagging="never"
-                   )
 demo.launch(show_api=False)