ehristoforu commited on
Commit
d8de0fb
•
1 Parent(s): 7c16112

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -82,9 +82,10 @@ def generate(
82
 
83
  pipe.to(device)
84
  seed = int(randomize_seed_fn(seed, randomize_seed))
85
-
86
  if not use_negative_prompt:
87
  negative_prompt = "" # type: ignore
 
88
  if model == "Alpha":
89
  images = pipe(
90
  prompt=prompt,
@@ -97,6 +98,7 @@ def generate(
97
  output_type="pil",
98
  ).images
99
  elif model == "Alpha inpaint":
 
100
  blurred_mask = pipe_inpaint.mask_processor.blur(mask_image, blur_factor=blur_factor)
101
  images = pipe_inpaint(
102
  prompt=prompt,
 
82
 
83
  pipe.to(device)
84
  seed = int(randomize_seed_fn(seed, randomize_seed))
85
+
86
  if not use_negative_prompt:
87
  negative_prompt = "" # type: ignore
88
+ images = None
89
  if model == "Alpha":
90
  images = pipe(
91
  prompt=prompt,
 
98
  output_type="pil",
99
  ).images
100
  elif model == "Alpha inpaint":
101
+ print("Inpainting...")
102
  blurred_mask = pipe_inpaint.mask_processor.blur(mask_image, blur_factor=blur_factor)
103
  images = pipe_inpaint(
104
  prompt=prompt,