Spaces:
Running
on
Zero
Running
on
Zero
Suchinthana
commited on
Commit
·
a5e00fd
1
Parent(s):
0b075c8
added diffusion strength and guidance_scale
Browse files
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(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|