Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -63,15 +63,11 @@ def generate_img(
|
|
| 63 |
sample_steps,
|
| 64 |
scale,
|
| 65 |
image_guidance_scale,
|
| 66 |
-
seed
|
| 67 |
-
|
| 68 |
-
pipeline: ImageEditor = None,
|
| 69 |
):
|
| 70 |
pil_image = PILImage.fromarray(np_image)
|
| 71 |
|
| 72 |
-
seed = int(randomize_seed_fn(seed, randomize_seed))
|
| 73 |
-
set_env(seed)
|
| 74 |
-
|
| 75 |
edited_image = pipeline.generate_edited_image(
|
| 76 |
instruction=edit_prompt,
|
| 77 |
conditioning_image=pil_image,
|
|
@@ -79,6 +75,7 @@ def generate_img(
|
|
| 79 |
num_inference_steps=sample_steps,
|
| 80 |
guidance_scale=scale,
|
| 81 |
image_guidance_scale=image_guidance_scale,
|
|
|
|
| 82 |
)[0]
|
| 83 |
result_image = np.asarray(edited_image)
|
| 84 |
return result_image
|
|
@@ -120,7 +117,6 @@ if __name__ == "__main__":
|
|
| 120 |
step=1,
|
| 121 |
value=42,
|
| 122 |
),
|
| 123 |
-
gr.Checkbox(label="Randomize seed", value=False),
|
| 124 |
],
|
| 125 |
outputs=[
|
| 126 |
Image(type="numpy", label="Img"),
|
|
|
|
| 63 |
sample_steps,
|
| 64 |
scale,
|
| 65 |
image_guidance_scale,
|
| 66 |
+
seed,
|
| 67 |
+
pipeline: ImageEditor,
|
|
|
|
| 68 |
):
|
| 69 |
pil_image = PILImage.fromarray(np_image)
|
| 70 |
|
|
|
|
|
|
|
|
|
|
| 71 |
edited_image = pipeline.generate_edited_image(
|
| 72 |
instruction=edit_prompt,
|
| 73 |
conditioning_image=pil_image,
|
|
|
|
| 75 |
num_inference_steps=sample_steps,
|
| 76 |
guidance_scale=scale,
|
| 77 |
image_guidance_scale=image_guidance_scale,
|
| 78 |
+
seed=seed,
|
| 79 |
)[0]
|
| 80 |
result_image = np.asarray(edited_image)
|
| 81 |
return result_image
|
|
|
|
| 117 |
step=1,
|
| 118 |
value=42,
|
| 119 |
),
|
|
|
|
| 120 |
],
|
| 121 |
outputs=[
|
| 122 |
Image(type="numpy", label="Img"),
|