Update app.py

app.py

@@ -1,19 +1,20 @@
+!pip install torch transformers gradio
+
+import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-import gradio as gr
 
 # モデルとトークナイザーの読み込み
-def load_model():
-    model_name = "EleutherAI/pythia-1b-deduped"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-    return model, tokenizer
-
-model, tokenizer = load_model()
+model_name = "EleutherAI/pythia-1b-deduped"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # 応答を生成する関数
 def respond(message, history, max_tokens, temperature, top_p):
     # 入力履歴と新しいメッセージを連結
+    if history is None:
+        history = []
+
     input_text = ""
     for user_message, bot_response in history:
         input_text += f"User: {user_message}\nAssistant: {bot_response}\n"
@@ -55,7 +56,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])
+    send_button.click(respond, inputs=[msg, chatbot, max_tokens, temperature, top_p], outputs=[chatbot, chatbot], _js=None)
     clear.click(clear_history, outputs=[chatbot])
 
 demo.launch()