Update app.py

app.py

@@ -33,7 +33,7 @@ def run_model(input_ids, model, max_new_tokens, top_p, top_k, temperature, repet
 def generate_text(mode: str, message: str, chat_history: List[Tuple[str, str]], max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
                   temperature: float = 0.6, top_p: float = 0.7, top_k: int = 20, repetition_penalty: float = 1.0) -> Iterator[str]:
     conversation = [{"role": "user", "content": user} for user, _ in chat_history]
-    conversation.append({"role": "assistant", "content": assistant} for _, assistant in chat_history)
+    conversation.extend([{"role": "assistant", "content": assistant} for _, assistant in chat_history])
     conversation.append({"role": "user", "content": message})
 
     context = "\n".join(f"{entry['role']}: {entry['content']}" for entry in conversation)
@@ -62,7 +62,7 @@ with gr.Blocks() as demo:
         mode_selector = gr.Radio(["generate", "edit"], label="Mode", value="generate")
         input_text = gr.Textbox(label="Input Text")
         output_text = gr.Textbox(label="Output")
-        chat_history = gr.State(default=[])
+        chat_history = gr.State(default=[])  # Corrected 'default' keyword
 
     generate_button = gr.Button("Generate/Edit")
     generate_button.click(switch_mode, inputs=[mode_selector, input_text, chat_history], outputs=output_text)