alakxender commited on
Commit
ba9dade
·
1 Parent(s): cc2019d
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -101,7 +101,7 @@ with gr.Blocks(title="Dhivehi Image to Text",css=css) as demo:
101
  label="Temperature", info="Controls randomness in generation"
102
  )
103
  top_p_slider = gr.Slider(
104
- minimum=0.1, maximum=1.0, value=0.9, step=0.1,
105
  label="Top-p", info="Controls diversity via nucleus sampling"
106
  )
107
  repetition_penalty_slider = gr.Slider(
@@ -144,11 +144,12 @@ with gr.Blocks(title="Dhivehi Image to Text",css=css) as demo:
144
  generate_button.click(
145
  fn=process_image_multiline,
146
  inputs=[model_path_dropdown, image_input, temperature_slider, top_p_slider, repetition_penalty_slider],
147
- outputs=text_output
 
148
  )
149
 
150
  show_event = stream_button.click(fn=show_stop_button_image, outputs=[stop_button, stream_button, generate_button])
151
- gen_event = show_event.then(fn=process_image_multiline_stream, inputs=[model_path_dropdown, image_input, temperature_slider, top_p_slider, repetition_penalty_slider], outputs=text_output)
152
  gen_event.then(fn=hide_stop_button_image, outputs=[stop_button, stream_button, generate_button])
153
  stop_button.click(fn=hide_stop_button_image, outputs=[stop_button, stream_button, generate_button], cancels=[gen_event])
154
 
@@ -186,11 +187,12 @@ with gr.Blocks(title="Dhivehi Image to Text",css=css) as demo:
186
  pdf_generate_button.click(
187
  fn=process_pdf_multiline,
188
  inputs=[model_path_dropdown, pdf_input, temperature_slider, top_p_slider, repetition_penalty_slider],
189
- outputs=pdf_text_output
 
190
  )
191
 
192
  pdf_show_event = pdf_stream_button.click(fn=show_stop_button_pdf, outputs=[pdf_stop_button, pdf_stream_button, pdf_generate_button])
193
- pdf_gen_event = pdf_show_event.then(fn=process_pdf_multiline_stream, inputs=[model_path_dropdown, pdf_input, temperature_slider, top_p_slider, repetition_penalty_slider], outputs=pdf_text_output)
194
  pdf_gen_event.then(fn=hide_stop_button_pdf, outputs=[pdf_stop_button, pdf_stream_button, pdf_generate_button])
195
  pdf_stop_button.click(fn=hide_stop_button_pdf, outputs=[pdf_stop_button, pdf_stream_button, pdf_generate_button], cancels=[pdf_gen_event])
196
 
 
101
  label="Temperature", info="Controls randomness in generation"
102
  )
103
  top_p_slider = gr.Slider(
104
+ minimum=0.1, maximum=1.0, value=1, step=0.1,
105
  label="Top-p", info="Controls diversity via nucleus sampling"
106
  )
107
  repetition_penalty_slider = gr.Slider(
 
144
  generate_button.click(
145
  fn=process_image_multiline,
146
  inputs=[model_path_dropdown, image_input, temperature_slider, top_p_slider, repetition_penalty_slider],
147
+ outputs=text_output,
148
+ show_progress="full"
149
  )
150
 
151
  show_event = stream_button.click(fn=show_stop_button_image, outputs=[stop_button, stream_button, generate_button])
152
+ gen_event = show_event.then(fn=process_image_multiline_stream, inputs=[model_path_dropdown, image_input, temperature_slider, top_p_slider, repetition_penalty_slider], outputs=text_output, show_progress="full")
153
  gen_event.then(fn=hide_stop_button_image, outputs=[stop_button, stream_button, generate_button])
154
  stop_button.click(fn=hide_stop_button_image, outputs=[stop_button, stream_button, generate_button], cancels=[gen_event])
155
 
 
187
  pdf_generate_button.click(
188
  fn=process_pdf_multiline,
189
  inputs=[model_path_dropdown, pdf_input, temperature_slider, top_p_slider, repetition_penalty_slider],
190
+ outputs=pdf_text_output,
191
+ show_progress="full"
192
  )
193
 
194
  pdf_show_event = pdf_stream_button.click(fn=show_stop_button_pdf, outputs=[pdf_stop_button, pdf_stream_button, pdf_generate_button])
195
+ pdf_gen_event = pdf_show_event.then(fn=process_pdf_multiline_stream, inputs=[model_path_dropdown, pdf_input, temperature_slider, top_p_slider, repetition_penalty_slider], outputs=pdf_text_output, show_progress="full")
196
  pdf_gen_event.then(fn=hide_stop_button_pdf, outputs=[pdf_stop_button, pdf_stream_button, pdf_generate_button])
197
  pdf_stop_button.click(fn=hide_stop_button_pdf, outputs=[pdf_stop_button, pdf_stream_button, pdf_generate_button], cancels=[pdf_gen_event])
198