Spaces:
Runtime error
Runtime error
Fix
Browse files
app.py
CHANGED
|
@@ -58,7 +58,8 @@ style_list = [
|
|
| 58 |
]
|
| 59 |
|
| 60 |
styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
|
| 61 |
-
|
|
|
|
| 62 |
style_names = list(styles.keys())
|
| 63 |
|
| 64 |
|
|
@@ -99,18 +100,18 @@ def run(
|
|
| 99 |
image: PIL.Image.Image,
|
| 100 |
prompt: str,
|
| 101 |
negative_prompt: str,
|
| 102 |
-
|
| 103 |
-
num_steps=25,
|
| 104 |
-
guidance_scale=5,
|
| 105 |
-
adapter_conditioning_scale=0.8,
|
| 106 |
-
cond_tau=0.8,
|
| 107 |
-
seed=0,
|
| 108 |
) -> PIL.Image.Image:
|
| 109 |
image = image.convert("RGB").resize((1024, 1024))
|
| 110 |
image = TF.to_tensor(image) > 0.5
|
| 111 |
image = TF.to_pil_image(image.to(torch.float32))
|
| 112 |
|
| 113 |
-
prompt, negative_prompt = apply_style(
|
| 114 |
|
| 115 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 116 |
out = pipe(
|
|
@@ -144,7 +145,7 @@ with gr.Blocks() as demo:
|
|
| 144 |
prompt = gr.Textbox(label="Prompt")
|
| 145 |
run_button = gr.Button("Run")
|
| 146 |
with gr.Accordion("Advanced options", open=False):
|
| 147 |
-
style = gr.Dropdown(choices=style_names, value=
|
| 148 |
negative_prompt = gr.Textbox(label="Negative prompt", value="")
|
| 149 |
num_steps = gr.Slider(
|
| 150 |
label="Number of steps",
|
|
|
|
| 58 |
]
|
| 59 |
|
| 60 |
styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
|
| 61 |
+
default_style_name = "Photographic"
|
| 62 |
+
default_style = styles[default_style_name]
|
| 63 |
style_names = list(styles.keys())
|
| 64 |
|
| 65 |
|
|
|
|
| 100 |
image: PIL.Image.Image,
|
| 101 |
prompt: str,
|
| 102 |
negative_prompt: str,
|
| 103 |
+
style_name: str = default_style_name,
|
| 104 |
+
num_steps: int = 25,
|
| 105 |
+
guidance_scale: float = 5,
|
| 106 |
+
adapter_conditioning_scale: float = 0.8,
|
| 107 |
+
cond_tau: float = 0.8,
|
| 108 |
+
seed: int = 0,
|
| 109 |
) -> PIL.Image.Image:
|
| 110 |
image = image.convert("RGB").resize((1024, 1024))
|
| 111 |
image = TF.to_tensor(image) > 0.5
|
| 112 |
image = TF.to_pil_image(image.to(torch.float32))
|
| 113 |
|
| 114 |
+
prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
|
| 115 |
|
| 116 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 117 |
out = pipe(
|
|
|
|
| 145 |
prompt = gr.Textbox(label="Prompt")
|
| 146 |
run_button = gr.Button("Run")
|
| 147 |
with gr.Accordion("Advanced options", open=False):
|
| 148 |
+
style = gr.Dropdown(choices=style_names, value=default_style_name, label="Style")
|
| 149 |
negative_prompt = gr.Textbox(label="Negative prompt", value="")
|
| 150 |
num_steps = gr.Slider(
|
| 151 |
label="Number of steps",
|