update clear flag

app.py

@@ -44,16 +44,18 @@ def get_dark_mode():
     """
 
 def clear_session():
-    return gr.update(value=None), gr.update(value=None)
+    return 'None', 'None'
 
 
 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, clear_flag):
-
-    if options == 'GPU':
-        decorated_run = spaces.GPU(run_gpu)
-        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, clear_flag)
+    if img_init == 'None':
+        return gr.update(value=None), gr.update(value=None)
     else:
-        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, clear_flag)
+        if options == 'GPU':
+            decorated_run = spaces.GPU(run_gpu)
+            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, clear_flag)
+        else:
+            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, clear_flag)
 
 def read_temp_file(temp_file_wrapper):
     name = temp_file_wrapper.name