Tophness2022 commited on
Commit
e2c9211
·
1 Parent(s): 672e555

fix gallery disappearing while next queued item generating

Browse files
Files changed (1) hide show
  1. gradio_server.py +6 -5
gradio_server.py CHANGED
@@ -1176,7 +1176,12 @@ def build_callback(taskid, state, pipe, num_inference_steps, repeats):
1176
  return update_progress
1177
 
1178
  def refresh_gallery(state):
1179
- return state
 
 
 
 
 
1180
 
1181
  def finalize_gallery(state):
1182
  choice = 0
@@ -2318,10 +2323,6 @@ def generate_video_tab(image2video=False):
2318
  return gr.update(), gr.update(value=image_data_to_show), gr.update(visible=True)
2319
  else:
2320
  return gr.update(), gr.update(), gr.update(visible=False)
2321
- def refresh_gallery_on_trigger(state):
2322
- if(state.get("update_gallery", False)):
2323
- state['update_gallery'] = False
2324
- return gr.update(value=state.get("file_list", []))
2325
  selected_indices = gr.State([])
2326
  queue_df.select(
2327
  fn=handle_selection,
 
1176
  return update_progress
1177
 
1178
  def refresh_gallery(state):
1179
+ return gr.update(value=state.get("file_list", []))
1180
+
1181
+ def refresh_gallery_on_trigger(state):
1182
+ if(state.get("update_gallery", False)):
1183
+ state['update_gallery'] = False
1184
+ return gr.update(value=state.get("file_list", []))
1185
 
1186
  def finalize_gallery(state):
1187
  choice = 0
 
2323
  return gr.update(), gr.update(value=image_data_to_show), gr.update(visible=True)
2324
  else:
2325
  return gr.update(), gr.update(), gr.update(visible=False)
 
 
 
 
2326
  selected_indices = gr.State([])
2327
  queue_df.select(
2328
  fn=handle_selection,