Update app.py

app.py

@@ -12,8 +12,9 @@ def format_prompt(message, history):
     return prompt
 
 def generate(
-    prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
+    inputs, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
+    initial_prompt = inputs[0]
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -28,7 +29,7 @@ def generate(
         seed=42,
     )
 
-    formatted_prompt = format_prompt(prompt, history)
+    formatted_prompt = format_prompt(initial_prompt, history)
 
     stream = client.text_generation(
         formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False
@@ -82,13 +83,9 @@ additional_inputs = [
 initial_prompt = "Hello John! Welcome to our simulation interview for the Solution Architect position. I understand you may have some initial thoughts or questions about the job requirements. Feel free to share those with me."
 
 gr.ChatInterface(
-    generate,
-    title="""Gen-AIInterviewUseCase featuring Mistral 7B""",
-    inputs=[
-        gr.Textbox(
-            default=initial_prompt, label="Initial Prompt", type="text", key="prompt"
-        )
-    ],
-    outputs=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
+    fn=generate,
+    inputs=[gr.Textbox(default=initial_prompt, label="Initial Prompt", type="text")],
+    outputs=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, layout="panel"),
     additional_inputs=additional_inputs,
+    title="""Gen-AIInterviewUseCase featuring Mistral 7B""",
 ).launch(show_api=False)