jbilcke-hf HF staff commited on
Commit
5e6df01
1 Parent(s): a945d6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -191,7 +191,7 @@ def plot_boxes_to_image(image_pil, tgt):
191
  def load_image(image_path):
192
  # # load image
193
  if isinstance(image_path, PIL.Image.Image):
194
- image_pil = image_path
195
  else:
196
  image_pil = Image.open(image_path).convert("RGB") # load image
197
 
 
191
  def load_image(image_path):
192
  # # load image
193
  if isinstance(image_path, PIL.Image.Image):
194
+ image_pil = image_path.convert("RGB")
195
  else:
196
  image_pil = Image.open(image_path).convert("RGB") # load image
197