myn0908 commited on
Commit
92ce23d
1 Parent(s): 7b4b94c

fixing missing value

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -57,7 +57,7 @@ def clear_session(image):
57
  return gr.update(value=None), gr.update(value=None)
58
 
59
 
60
- 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):
61
  global clear_flag
62
  if clear_flag:
63
  clear_flag = False # Reset the flag after handling the clear action
@@ -65,9 +65,9 @@ def assign_gpu(options, img_init, text_init, prompt_template_init, style_name_in
65
  else:
66
  if options == 'GPU':
67
  decorated_run = spaces.GPU(run_gpu)
68
- return decorated_run(options, img_init, text_init, prompt_template_init, style_name_init, seeds_init, val_r_values_init, faster_init, model_name_init, input_type)
69
  else:
70
- return run_cpu(options, img_init, text_init, prompt_template_init, style_name_init, seeds_init, val_r_values_init, faster_init, model_name_init, input_type)
71
 
72
  def read_temp_file(temp_file_wrapper):
73
  name = temp_file_wrapper.name
 
57
  return gr.update(value=None), gr.update(value=None)
58
 
59
 
60
+ 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, prompt_quality):
61
  global clear_flag
62
  if clear_flag:
63
  clear_flag = False # Reset the flag after handling the clear action
 
65
  else:
66
  if options == 'GPU':
67
  decorated_run = spaces.GPU(run_gpu)
68
+ return decorated_run(options, img_init, text_init, prompt_template_init, style_name_init, seeds_init, val_r_values_init, faster_init, model_name_init, input_type, prompt_quality)
69
  else:
70
+ return run_cpu(options, img_init, text_init, prompt_template_init, style_name_init, seeds_init, val_r_values_init, faster_init, model_name_init, input_type, prompt_quality)
71
 
72
  def read_temp_file(temp_file_wrapper):
73
  name = temp_file_wrapper.name