Update app.py

app.py

@@ -43,7 +43,7 @@ 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,
+    message, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -59,7 +59,7 @@ def generate(
         seed=42,
     )
 
-    formatted_prompt = format_prompt(prompt, history)
+    formatted_prompt = format_prompt(message, history)
 
     stream = client.text_generation(
         formatted_prompt,
@@ -116,7 +116,8 @@ additional_inputs = [
 
 gr.ChatInterface(
     fn=generate,
-    chatbot=gr.Chatbot(
+    inputs=[gr.Textbox(label="Interviewer", type="text", default=initial_prompt)],
+    outputs=gr.Chatbot(
         show_label=False,
         show_share_button=False,
         show_copy_button=True,