2z299 commited on
Commit
d4fa487
·
verified ·
1 Parent(s): 247dc3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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,