craftgamesnetwork commited on
Commit
22b3f33
·
verified ·
1 Parent(s): 4548375

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -114,7 +114,7 @@ def generate(
114
  negative_prompt_2 = None # type: ignore
115
 
116
  if use_inpainting:
117
- images = pipe(
118
  prompt=prompt,
119
  image_init=img_url,
120
  mask_image=mask_url,
@@ -126,7 +126,7 @@ def generate(
126
  num_inference_steps=num_inference_steps_base,
127
  generator=generator,
128
  ).images[0]
129
- return images
130
  elif use_img2img:
131
  images = pipe(
132
  prompt=prompt,
 
114
  negative_prompt_2 = None # type: ignore
115
 
116
  if use_inpainting:
117
+ image = pipe(
118
  prompt=prompt,
119
  image_init=img_url,
120
  mask_image=mask_url,
 
126
  num_inference_steps=num_inference_steps_base,
127
  generator=generator,
128
  ).images[0]
129
+ return image
130
  elif use_img2img:
131
  images = pipe(
132
  prompt=prompt,