Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -27,8 +27,8 @@ def get_model_and_tokenizer(model_name):
|
|
27 |
return model, tokenizer
|
28 |
|
29 |
def disable_generate_button():
|
30 |
-
#
|
31 |
-
return gr.update(interactive=False,
|
32 |
|
33 |
def load_model(model_name):
|
34 |
"""
|
@@ -40,8 +40,8 @@ def load_model(model_name):
|
|
40 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HF_TOKEN)
|
41 |
loaded_models[model_name] = (model, tokenizer)
|
42 |
status_message = f"Model '{model_name}' loaded successfully."
|
43 |
-
#
|
44 |
-
return status_message, gr.update(interactive=True,
|
45 |
|
46 |
@spaces.GPU
|
47 |
def generate_text(
|
@@ -129,7 +129,7 @@ with gr.Blocks() as demo:
|
|
129 |
generate_btn = gr.Button("続きを生成", variant="primary")
|
130 |
clear_btn = gr.Button("クリア")
|
131 |
|
132 |
-
#
|
133 |
model_dropdown.change(
|
134 |
fn=disable_generate_button,
|
135 |
inputs=None,
|
|
|
27 |
return model, tokenizer
|
28 |
|
29 |
def disable_generate_button():
|
30 |
+
# 生成ボタンを無効化し、テキストを「モデルをロード中……」に変更する
|
31 |
+
return gr.update(interactive=False, value="モデルをロード中……")
|
32 |
|
33 |
def load_model(model_name):
|
34 |
"""
|
|
|
40 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HF_TOKEN)
|
41 |
loaded_models[model_name] = (model, tokenizer)
|
42 |
status_message = f"Model '{model_name}' loaded successfully."
|
43 |
+
# ロード完了後、生成ボタンを有効化し、テキストを「続きを生成」に戻す
|
44 |
+
return status_message, gr.update(interactive=True, value="続きを生成")
|
45 |
|
46 |
@spaces.GPU
|
47 |
def generate_text(
|
|
|
129 |
generate_btn = gr.Button("続きを生成", variant="primary")
|
130 |
clear_btn = gr.Button("クリア")
|
131 |
|
132 |
+
# プルダウン変更時に、まず生成ボタンを無効化(テキストを「モデルをロード中……」に変更)し、その後モデルをロードして生成ボタンを再有効化するイベントチェーンを設定
|
133 |
model_dropdown.change(
|
134 |
fn=disable_generate_button,
|
135 |
inputs=None,
|