Fabrice-TIERCELIN commited on
Commit
b407674
·
verified ·
1 Parent(s): 4ff8938
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -158,7 +158,7 @@ def stage2_process_example(*args, **kwargs):
158
  output_filename = os.path.join(outputs_folder, datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + '.png')
159
  print(output_filename)
160
  iio.imwrite(output_filename, result_slider[1], format="png")
161
- return [gr.update(visible = True, value=output_filename), warning, dummy_button]
162
 
163
  def stage2_process(*args, **kwargs):
164
  try:
@@ -728,7 +728,8 @@ with gr.Blocks(js=js) as interface:
728
  outputs = [
729
  result_example,
730
  warning,
731
- dummy_button
 
732
  ],
733
  cache_examples = True,
734
  )
 
158
  output_filename = os.path.join(outputs_folder, datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + '.png')
159
  print(output_filename)
160
  iio.imwrite(output_filename, result_slider[1], format="png")
161
+ return [gr.update(visible = True, value=output_filename), warning, dummy_button, gr.skip()]
162
 
163
  def stage2_process(*args, **kwargs):
164
  try:
 
728
  outputs = [
729
  result_example,
730
  warning,
731
+ dummy_button,
732
+ prompt_hint
733
  ],
734
  cache_examples = True,
735
  )