TheOneHong commited on
Commit
e765d2d
·
1 Parent(s): 12af388
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,7 +47,7 @@ def generate_image(api_key, prompt, image_size, seed, sync_mode, num_images, ena
47
  except Exception as e:
48
  error_msg = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
49
  print(error_msg)
50
- return [gr.update(visible=False), gr.update(visible=False)]
51
 
52
  def update_safety_tolerance_visibility(enable_safety):
53
  return gr.update(visible=enable_safety, value="6")
@@ -82,7 +82,7 @@ with gr.Blocks() as demo:
82
 
83
  generate_btn = gr.Button("Generate Image")
84
  output_gallery = gr.Gallery(label="Generated Images", columns=2, rows=2)
85
- response_output = gr.Textbox(label="Response", visible=False)
86
 
87
  enable_safety_checker.change(
88
  fn=update_safety_tolerance_visibility,
 
47
  except Exception as e:
48
  error_msg = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
49
  print(error_msg)
50
+ return [gr.update(value=[]), gr.update(value=error_msg)]
51
 
52
  def update_safety_tolerance_visibility(enable_safety):
53
  return gr.update(visible=enable_safety, value="6")
 
82
 
83
  generate_btn = gr.Button("Generate Image")
84
  output_gallery = gr.Gallery(label="Generated Images", columns=2, rows=2)
85
+ response_output = gr.Textbox(label="Response", visible=True)
86
 
87
  enable_safety_checker.change(
88
  fn=update_safety_tolerance_visibility,