Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
 
+
 def format_prompt(message, history):
     prompt = "<s>"
     for user_prompt, bot_response in history:
@@ -11,6 +12,7 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
+
 def generate(
     prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
@@ -31,7 +33,11 @@ def generate(
     formatted_prompt = format_prompt(prompt, history)
 
     stream = client.text_generation(
-        formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False
+        formatted_prompt,
+        **generate_kwargs,
+        stream=True,
+        details=True,
+        return_full_text=False,
     )
     output = ""
 
@@ -40,6 +46,7 @@ def generate(
         yield output
     return output
 
+
 additional_inputs = [
     gr.Slider(
         label="Temperature",
@@ -83,8 +90,14 @@ initial_prompt = "Hello John! Welcome to our simulation interview for the Soluti
 
 gr.ChatInterface(
     generate,
-    gr.Textbox(label="Initial Prompt", type="text", default=initial_prompt),
-    *additional_inputs,
-    gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
-    title="Gen-AIInterviewUseCase featuring Mistral 7B",
+    inputs=[gr.Textbox(label="Initial Prompt", type="text", default=initial_prompt)],
+    outputs=gr.Chatbot(
+        show_label=False,
+        show_share_button=False,
+        show_copy_button=True,
+        likeable=True,
+        layout="panel",
+    ),
+    additional_inputs=additional_inputs,
+    title="Mistral 7B",
 ).launch(show_api=False)