myn0908 commited on
Commit
f66f514
1 Parent(s): 2a9dcb2

update code

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -47,10 +47,8 @@ def get_dark_mode():
47
  def clear_session(image):
48
  global clear_flag
49
  clear_flag = True
50
- if image is not None:
51
- ones = Image.new("L", (1024, 1024), 255)
52
- return ones, gr.update(value=None)
53
- return gr.update(value=None), gr.update(value=None)
54
 
55
 
56
  def assign_gpu(options, img_init, text_init, prompt_template_init, style_name_init, seeds_init, val_r_values_init, faster_init, model_name_init, input_type):
 
47
  def clear_session(image):
48
  global clear_flag
49
  clear_flag = True
50
+ ones = Image.new("L", (1024, 1024), 255)
51
+ return ones, gr.update(value=None)
 
 
52
 
53
 
54
  def assign_gpu(options, img_init, text_init, prompt_template_init, style_name_init, seeds_init, val_r_values_init, faster_init, model_name_init, input_type):