fix

app.py

@@ -32,7 +32,7 @@ if torch.cuda.is_available():
 @torch.inference_mode()
 def generate(
     message: str,
-    chat_history: list[tuple[str, str]],
+    chat_history,
     max_new_tokens: int = 1024,
     temperature: float = 0.7,
     top_p: float = 0.95,
@@ -56,7 +56,7 @@ def generate(
         repetition_penalty=repetition_penalty,
         top_k=top_k,
         top_p=top_p,
-    )
+    )[-1]["generated_text"]
     print(output)
     gen_text = output[len(user_input):]
     gen_text = gen_text[:gen_text.find("\n")] if "\n" in gen_text else gen_text
@@ -65,7 +65,7 @@ def generate(
 
 demo = gr.ChatInterface(
     fn=generate,
-    type="tuples",
+    type="messages",
     additional_inputs_accordion=gr.Accordion(label="詳細設定", open=False),
     additional_inputs=[
         gr.Slider(