Update app.py

app.py

@@ -37,9 +37,9 @@ MODEL_NAME_MAP = {
     "150m-instruct3": "llm-jp/llm-jp-3-150m-instruct3",
     "440m-instruct3": "llm-jp/llm-jp-3-440m-instruct3",
     "980m-instruct3": "llm-jp/llm-jp-3-980m-instruct3",
-    "1.8b-instruct3": "llm-jp/llm-jp-3-1.8b-instruct3",
-    "3.7b-instruct3": "llm-jp/llm-jp-3-3.7b-instruct3",
-    "13b-instruct3": "llm-jp/llm-jp-3-13b-instruct3",
+    # "1.8b-instruct3": "llm-jp/llm-jp-3-1.8b-instruct3",
+    # "3.7b-instruct3": "llm-jp/llm-jp-3-3.7b-instruct3",
+    # "13b-instruct3": "llm-jp/llm-jp-3-13b-instruct3",
 }
 
 quantization_config = BitsAndBytesConfig(
@@ -50,11 +50,11 @@ quantization_config = BitsAndBytesConfig(
 )
 MODELS = {
     key: AutoModelForCausalLM.from_pretrained(
-        MODEL_NAME, quantization_config=quantization_config, device_map="auto"
-    ) for key, value in MODEL_NAME_MAP.items()
+        repo_id, quantization_config=quantization_config, device_map="auto"
+    ) for key, repo_id in MODEL_NAME_MAP.items()
 }
 TOKENIZERS = {
-    key: AutoTokenizer.from_pretrained(MODEL_NAME) for key, value in MODEL_NAME_MAP.items()
+    key: AutoTokenizer.from_pretrained(repo_id) for key, repo_id in MODEL_NAME_MAP.items()
     
 }
 
@@ -189,21 +189,17 @@ def demo():
                 components=[chat_history], value="🗑️ 削除", scale=1,
             )
 
-        with gr.Group():
-            with gr.Row():
-                input_text = gr.Textbox(
-                    value="",
-                    placeholder="質問を入力してください...",
-                    show_label=False,
-                    scale=8,
-                )
-                start_btn = gr.Button(
-                    value="送信",
-                    variant="primary",
-                    scale=2,
-                )
-            gr.Markdown(
-                value="※ 誤った情報を生成する可能性があります。"
+        with gr.Row():
+            input_text = gr.Textbox(
+                value="",
+                placeholder="質問を入力してください...",
+                show_label=False,
+                scale=8,
+            )
+            start_btn = gr.Button(
+                value="送信",
+                variant="primary",
+                scale=2,
             )
 
         with gr.Accordion(label="詳細設定", open=False):