SkalskiP commited on
Commit
867296e
1 Parent(s): 3e075bb

small mask fix

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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=1024,
42
- height=1024,
43
- strength=0.9
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