Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 # モデルとトークナイザーの読み込み
 model_name = "EleutherAI/pythia-1b-deduped"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name, ignore_mismatched_sizes=True)
 
 # 応答を生成する関数
 def respond(message, history, max_tokens, temperature, top_p):
@@ -54,7 +54,7 @@ with gr.Blocks() as demo:
     def clear_history():
         return [], []
 
-    send_button.click(respond, inputs=[msg, chatbot, max_tokens, temperature, top_p], outputs=[chatbot, chatbot], _js=None)
+    send_button.click(respond, inputs=[msg, chatbot, max_tokens, temperature, top_p], outputs=[chatbot, chatbot])
     clear.click(clear_history, outputs=[chatbot])
 
 demo.launch()