2z299 commited on
Commit
4ab04be
·
verified ·
1 Parent(s): e4d2611

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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, label="モデルをロード中……")
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, label="続きを生成")
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,