Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
2c9f146
1
Parent(s):
10bdb89
update app.py
Browse files
app.py
CHANGED
|
@@ -154,12 +154,12 @@ def generate_image(
|
|
| 154 |
enable_anti_blur,
|
| 155 |
model_version
|
| 156 |
):
|
| 157 |
-
|
|
|
|
| 158 |
|
| 159 |
-
|
| 160 |
-
|
| 161 |
|
| 162 |
-
try:
|
| 163 |
image = pipeline(
|
| 164 |
id_image=input_image,
|
| 165 |
prompt=prompt,
|
|
@@ -178,7 +178,7 @@ def generate_image(
|
|
| 178 |
gr.Error(f"An error occurred: {e}")
|
| 179 |
return gr.update()
|
| 180 |
|
| 181 |
-
return gr.update(value
|
| 182 |
|
| 183 |
|
| 184 |
def generate_examples(id_image, control_image, prompt_text, seed, enable_realism, enable_anti_blur, model_version):
|
|
|
|
| 154 |
enable_anti_blur,
|
| 155 |
model_version
|
| 156 |
):
|
| 157 |
+
try:
|
| 158 |
+
pipeline = prepare_pipeline(model_version=model_version, enable_realism=enable_realism, enable_anti_blur=enable_anti_blur)
|
| 159 |
|
| 160 |
+
if seed == 0:
|
| 161 |
+
seed = torch.seed() & 0xFFFFFFFF
|
| 162 |
|
|
|
|
| 163 |
image = pipeline(
|
| 164 |
id_image=input_image,
|
| 165 |
prompt=prompt,
|
|
|
|
| 178 |
gr.Error(f"An error occurred: {e}")
|
| 179 |
return gr.update()
|
| 180 |
|
| 181 |
+
return gr.update(value=image, label=f"Generated image, seed = {seed}")
|
| 182 |
|
| 183 |
|
| 184 |
def generate_examples(id_image, control_image, prompt_text, seed, enable_realism, enable_anti_blur, model_version):
|