RageshAntony commited on
Commit
ba278c4
·
verified ·
1 Parent(s): 5e26a4d
Files changed (1) hide show
  1. app.py +17 -17
app.py CHANGED
@@ -387,23 +387,23 @@ with gr.Blocks(css=css) as demo:
387
  with gr.Tab(model_name):
388
  results[model_name] = gr.Image(label=f"{model_name} Result")
389
  seeds[model_name] = gr.Number(label="Seed used", visible=True)
390
- with gr.Column(scale=1):
391
- gr.Markdown("### Generated Images")
392
- file_gallery = gr.Gallery(
393
- label="Generated Images",
394
- show_label=False,
395
- elem_id="file_gallery",
396
- columns=3,
397
- height=800,
398
- visible=True
399
- )
400
- refresh_button = gr.Button("Refresh Gallery",visible=True)
401
-
402
- refresh_button.click(
403
- fn=update_gallery,
404
- inputs=[],
405
- outputs=[file_gallery],
406
- )
407
 
408
  output_components = []
409
  for model_name in MODEL_CONFIGS.keys():
 
387
  with gr.Tab(model_name):
388
  results[model_name] = gr.Image(label=f"{model_name} Result")
389
  seeds[model_name] = gr.Number(label="Seed used", visible=True)
390
+ with gr.Column(scale=1):
391
+ gr.Markdown("### Generated Images")
392
+ file_gallery = gr.Gallery(
393
+ label="Generated Images",
394
+ show_label=False,
395
+ elem_id="file_gallery",
396
+ columns=3,
397
+ height=800,
398
+ visible=True
399
+ )
400
+ refresh_button = gr.Button("Refresh Gallery",visible=True)
401
+
402
+ refresh_button.click(
403
+ fn=update_gallery,
404
+ inputs=[tabs],
405
+ outputs=[file_gallery],
406
+ )
407
 
408
  output_components = []
409
  for model_name in MODEL_CONFIGS.keys():