GiorgioV commited on
Commit
08503ba
·
verified ·
1 Parent(s): 5555e77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -197,11 +197,11 @@ def infer(
197
  path_or_img = item
198
 
199
  if isinstance(path_or_img, str):
200
- pil_images.append(Image.open(path_or_img).convert("RGB"))
201
  elif isinstance(path_or_img, Image.Image):
202
- pil_images.append(path_or_img.convert("RGB"))
203
  else:
204
- pil_images.append(Image.open(path_or_img.name).convert("RGB"))
205
  except Exception as e:
206
  print(f"Skipping invalid image item: {e}")
207
  continue
@@ -290,7 +290,7 @@ def infer(
290
  generator=generator,
291
  true_cfg_scale=guidance_scale,
292
  ).images[0]
293
-
294
  return result_image, seed
295
 
296
  except Exception as e:
 
197
  path_or_img = item
198
 
199
  if isinstance(path_or_img, str):
200
+ pil_images.append(Image.open(path_or_img))#.convert("RGB"))
201
  elif isinstance(path_or_img, Image.Image):
202
+ pil_images.append(path_or_img)#.convert("RGB"))
203
  else:
204
+ pil_images.append(Image.open(path_or_img.name))#.convert("RGB"))
205
  except Exception as e:
206
  print(f"Skipping invalid image item: {e}")
207
  continue
 
290
  generator=generator,
291
  true_cfg_scale=guidance_scale,
292
  ).images[0]
293
+ result_image = result_image.convert("RGB")
294
  return result_image, seed
295
 
296
  except Exception as e: