Suchinthana commited on
Commit
a5e00fd
·
1 Parent(s): 0b075c8

added diffusion strength and guidance_scale

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -154,7 +154,14 @@ def make_inpaint_condition(init_image, mask_image):
154
  @spaces.GPU
155
  def generate_satellite_image(init_image, mask_image, prompt):
156
  control_image = make_inpaint_condition(init_image, mask_image)
157
- result = pipeline(prompt=prompt, image=init_image, mask_image=mask_image, control_image=control_image)
 
 
 
 
 
 
 
158
  return result.images[0]
159
 
160
  # Gradio UI
 
154
  @spaces.GPU
155
  def generate_satellite_image(init_image, mask_image, prompt):
156
  control_image = make_inpaint_condition(init_image, mask_image)
157
+ result = pipeline(
158
+ prompt=prompt,
159
+ image=init_image,
160
+ mask_image=mask_image,
161
+ control_image=control_image,
162
+ strength=0.65,
163
+ guidance_scale=85
164
+ )
165
  return result.images[0]
166
 
167
  # Gradio UI