fffiloni commited on
Commit
0243396
·
1 Parent(s): dcd9004

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -302,8 +302,8 @@ def infer(video_in, trim_value, prompt, background_prompt, background_type, box_
302
 
303
  matte_img = run_grounded_sam(image_array, prompt, "text", background_prompt, background_type, box_threshold, text_threshold, iou_threshold, scribble_mode, guidance_mode)
304
  #print(pix2pix_img)u
305
- matte_img = Image.open(matte_img)
306
- #rgb_im = image.convert("RGB")
307
 
308
  # exporting the image
309
  matte_img.save(f"result_img-{i}.jpg")
 
302
 
303
  matte_img = run_grounded_sam(image_array, prompt, "text", background_prompt, background_type, box_threshold, text_threshold, iou_threshold, scribble_mode, guidance_mode)
304
  #print(pix2pix_img)u
305
+ matte_img = Image.fromarray(matte_img)
306
+ #rgb_im = image.convertt("RGB")
307
 
308
  # exporting the image
309
  matte_img.save(f"result_img-{i}.jpg")