Spaces:
Runtime error
Runtime error
Bugifx for input type
Browse files
app.py
CHANGED
@@ -51,14 +51,15 @@ def infer(prompt, steps, scale, seed, editing_prompt_1 = None, reverse_editing_d
|
|
51 |
del edit_threshold[index]
|
52 |
editing_prompt_embeddings = None
|
53 |
|
54 |
-
|
55 |
-
if edit_style is not None:
|
56 |
editing_prompt = None
|
57 |
reverse_editing_direction = reverse_editing_direction_style
|
58 |
edit_warmup_steps = edit_warmup_steps_style
|
59 |
edit_guidance_scale = edit_guidance_scale_style
|
60 |
edit_threshold = edit_threshold_style
|
61 |
editing_prompt_embeddings = style_embeddings[edit_style]
|
|
|
62 |
|
63 |
gen.manual_seed(seed)
|
64 |
images.extend(pipe(prompt, guidance_scale=scale, num_inference_steps=steps, generator=gen,
|
@@ -67,7 +68,7 @@ def infer(prompt, steps, scale, seed, editing_prompt_1 = None, reverse_editing_d
|
|
67 |
edit_momentum_scale=edit_momentum_scale, edit_mom_beta=edit_mom_beta
|
68 |
).images)
|
69 |
|
70 |
-
return zip(images, ['Original',
|
71 |
|
72 |
def reset_style():
|
73 |
radio = gr.Radio(label='Style', choices=['Concept Art', 'Animation', 'Character Design', 'Portrait Photo', 'Architecture'])
|
|
|
51 |
del edit_threshold[index]
|
52 |
editing_prompt_embeddings = None
|
53 |
|
54 |
+
out_label = 'SEGA'
|
55 |
+
if edit_style is not None and isinstance(edit_style, str) and edit_style in style_embeddings.keys():
|
56 |
editing_prompt = None
|
57 |
reverse_editing_direction = reverse_editing_direction_style
|
58 |
edit_warmup_steps = edit_warmup_steps_style
|
59 |
edit_guidance_scale = edit_guidance_scale_style
|
60 |
edit_threshold = edit_threshold_style
|
61 |
editing_prompt_embeddings = style_embeddings[edit_style]
|
62 |
+
out_label = edit_style
|
63 |
|
64 |
gen.manual_seed(seed)
|
65 |
images.extend(pipe(prompt, guidance_scale=scale, num_inference_steps=steps, generator=gen,
|
|
|
68 |
edit_momentum_scale=edit_momentum_scale, edit_mom_beta=edit_mom_beta
|
69 |
).images)
|
70 |
|
71 |
+
return zip(images, ['Original', out_label])
|
72 |
|
73 |
def reset_style():
|
74 |
radio = gr.Radio(label='Style', choices=['Concept Art', 'Animation', 'Character Design', 'Portrait Photo', 'Architecture'])
|