Update app.py

app.py

@@ -22,6 +22,8 @@ import gradio as gr
 import argparse
 
 import numpy as np
+from PIL import image
+from moviepy.editor import *
 import torch
 from torch.nn import functional as F
 import torchvision
@@ -290,10 +292,13 @@ def infer(prompt,video_in, trim_value):
     print("set stop frames to: " + str(n_frame))
     
     for i in frames_list[0:int(n_frame)]:
-        numpy_i = Image.open(i).convert("RGB")
+        to_numpy_i = Image.open(i).convert("RGB")
         #need to convert to numpy
+        # Convert the image to a NumPy array
+        image_array = np.array(to_numpy_i)
+
         
-        matte_img = run_grounded_sam(numpy, text_prompt, task_type, background_prompt, background_type, box_threshold, text_threshold, iou_threshold, scribble_mode, guidance_mode):
+        matte_img = run_grounded_sam(image_array, text_prompt, task_type, background_prompt, background_type, box_threshold, text_threshold, iou_threshold, scribble_mode, guidance_mode):
         #print(pix2pix_img)
         #image = Image.open(pix2pix_img)
         #rgb_im = image.convert("RGB")