Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # モデルとトークナイザーの読み込み
-model_name = "Sakalti/iturkaAI-large"
+model_name = "Sakalti/gpt-neox-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, ignore_mismatched_sizes=True)
 
@@ -45,10 +45,10 @@ with gr.Blocks() as demo:
     gr.Markdown("## AIチャット")
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="あなたのメッセージ", placeholder="ここにメッセージを入力...")
-    max_tokens = gr.Slider(1, 2048, value=512, step=1, label="Max new tokens")
-    temperature = gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature")
-    top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
-    send_button = gr.Button("送信")
+    max_tokens = gr.Slider(1, 2048, value=512, step=1, label="新規トークン最大")
+    temperature = gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="温度")
+    top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p (核サンプリング)")
+    send_button = gr.Button("送さんむ信")
     clear = gr.Button("クリア")
 
     def clear_history():