Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def disable_generate_button():
|
|
32 |
|
33 |
def load_model(model_name):
|
34 |
"""
|
35 |
-
|
36 |
"""
|
37 |
tokenizer = AutoTokenizer.from_pretrained(
|
38 |
model_name, attn_implementation="flash_attention_2", use_auth_token=HF_TOKEN
|
@@ -140,6 +140,9 @@ with gr.Blocks() as demo:
|
|
140 |
outputs=[load_status, generate_btn]
|
141 |
)
|
142 |
|
|
|
|
|
|
|
143 |
# 生成ボタン押下時のイベント設定
|
144 |
generate_btn.click(
|
145 |
fn=generate_text,
|
|
|
32 |
|
33 |
def load_model(model_name):
|
34 |
"""
|
35 |
+
プルダウン変更時や起動時に呼ばれ、モデルをロードして生成ボタンを有効化する。
|
36 |
"""
|
37 |
tokenizer = AutoTokenizer.from_pretrained(
|
38 |
model_name, attn_implementation="flash_attention_2", use_auth_token=HF_TOKEN
|
|
|
140 |
outputs=[load_status, generate_btn]
|
141 |
)
|
142 |
|
143 |
+
# 起動時にも load_model を実行する(初期値のモデルでロード)
|
144 |
+
demo.load(fn=load_model, inputs=model_dropdown, outputs=[load_status, generate_btn])
|
145 |
+
|
146 |
# 生成ボタン押下時のイベント設定
|
147 |
generate_btn.click(
|
148 |
fn=generate_text,
|