Spaces:
Running
on
Zero
Running
on
Zero
update clear flag
Browse files
app.py
CHANGED
@@ -44,16 +44,18 @@ def get_dark_mode():
|
|
44 |
"""
|
45 |
|
46 |
def clear_session():
|
47 |
-
return
|
48 |
|
49 |
|
50 |
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):
|
51 |
-
|
52 |
-
|
53 |
-
decorated_run = spaces.GPU(run_gpu)
|
54 |
-
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)
|
55 |
else:
|
56 |
-
|
|
|
|
|
|
|
|
|
57 |
|
58 |
def read_temp_file(temp_file_wrapper):
|
59 |
name = temp_file_wrapper.name
|
|
|
44 |
"""
|
45 |
|
46 |
def clear_session():
|
47 |
+
return 'None', 'None'
|
48 |
|
49 |
|
50 |
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):
|
51 |
+
if img_init == 'None':
|
52 |
+
return gr.update(value=None), gr.update(value=None)
|
|
|
|
|
53 |
else:
|
54 |
+
if options == 'GPU':
|
55 |
+
decorated_run = spaces.GPU(run_gpu)
|
56 |
+
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)
|
57 |
+
else:
|
58 |
+
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)
|
59 |
|
60 |
def read_temp_file(temp_file_wrapper):
|
61 |
name = temp_file_wrapper.name
|