sawac commited on
Commit
a7cb98d
·
verified ·
1 Parent(s): e4456e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -32,7 +32,7 @@ def load_model():
32
 
33
  def get_loading_status():
34
  global loading_progress
35
- return loading_progress
36
 
37
  def get_llama_response(prompt):
38
  global llm, model_loaded
@@ -71,10 +71,11 @@ with gr.Blocks() as demo:
71
  output_text = gr.Textbox(label="Generated Response")
72
  submit_button = gr.Button("Submit")
73
 
74
- loading_bar = gr.progressbar(label="Model Loading Progress", max=100)
 
75
 
76
  submit_button.click(fn=greet, inputs=[input_text, intensity], outputs=output_text)
77
- demo.load(fn=get_loading_status, outputs=loading_bar, every=1)
78
 
79
  demo.queue()
80
  if __name__ == "__main__":
 
32
 
33
  def get_loading_status():
34
  global loading_progress
35
+ return loading_progress, f"モデル読み込み進捗: {loading_progress}%"
36
 
37
  def get_llama_response(prompt):
38
  global llm, model_loaded
 
71
  output_text = gr.Textbox(label="Generated Response")
72
  submit_button = gr.Button("Submit")
73
 
74
+ loading_progress = gr.Progress()
75
+ loading_status = gr.Textbox(label="Loading Status")
76
 
77
  submit_button.click(fn=greet, inputs=[input_text, intensity], outputs=output_text)
78
+ demo.load(fn=get_loading_status, outputs=[loading_progress, loading_status], every=1)
79
 
80
  demo.queue()
81
  if __name__ == "__main__":