small mask fix
Browse files
app.py
CHANGED
@@ -34,13 +34,15 @@ def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True
|
|
34 |
gr.Info("Please draw a mask on the image.")
|
35 |
return None
|
36 |
|
|
|
|
|
37 |
return pipe(
|
38 |
prompt=input_text,
|
39 |
image=image,
|
40 |
mask_image=mask_image,
|
41 |
-
width=
|
42 |
-
height=
|
43 |
-
strength=0.
|
44 |
).images[0]
|
45 |
|
46 |
|
|
|
34 |
gr.Info("Please draw a mask on the image.")
|
35 |
return None
|
36 |
|
37 |
+
width, height = image.size
|
38 |
+
|
39 |
return pipe(
|
40 |
prompt=input_text,
|
41 |
image=image,
|
42 |
mask_image=mask_image,
|
43 |
+
width=width,
|
44 |
+
height=height,
|
45 |
+
strength=0.7
|
46 |
).images[0]
|
47 |
|
48 |
|