yizhangliu commited on
Commit
856d473
·
1 Parent(s): 47f2583

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -188,8 +188,8 @@ def read_content(file_path: str) -> str:
188
  return content
189
 
190
  def predict(dict, prompt=""):
191
- init_image = dict["image"].convert("RGB") #.resize((512, 512))
192
- mask = dict["mask"].convert("RGB") #.resize((512, 512))
193
  output = process(init_image, mask)
194
  # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
195
 
 
188
  return content
189
 
190
  def predict(dict, prompt=""):
191
+ init_image = dict["image"] # .convert("RGB") #.resize((512, 512))
192
+ mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
193
  output = process(init_image, mask)
194
  # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
195