Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import yt_dlp
 import os 
 import json
-import rembg
+from rembg import remove as rm
 import cv2
 import uuid
 uid=uuid.uuid4()
@@ -115,7 +115,7 @@ def rem_bg(vid,frame_count):
         capture.set(cv2.CAP_PROP_POS_FRAMES, i)
         ret, frame_f = capture.read(i)
         frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
-        out = rembg(frame_ff)
+        out = rm(frame_ff)
 
         cv2.imwrite(f'{uid}-rembg/{i}.png',out)
         fbox2.append(f'{uid}-rembg/{i}.png')