iitolstykh commited on
Commit
eb3d4ce
·
verified ·
1 Parent(s): c287df1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -63,15 +63,11 @@ def generate_img(
63
  sample_steps,
64
  scale,
65
  image_guidance_scale,
66
- seed=0,
67
- randomize_seed=False,
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"),