Update README.md

README.md

@@ -109,7 +109,7 @@ import torch
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-file = "./image.png" # input image
+video_path = "/path_to_your_video.mp4"# input video
 
 model = BEN2.BEN_Base().to(device).eval() #init pipeline
 
@@ -119,7 +119,7 @@ model.loadcheckpoints("./BEN2_Base.pth")
 
 
 model.segment_video(
-    video_path="/path_to_your_video.mp4",
+    video_path= video_path,
     output_path="./", # Outputs will be saved as foreground.webm or foreground.mp4. The default value is "./"
     fps=0, # If this is set to 0 CV2 will detect the fps in the original video. The default value is 0.
     refine_foreground=False,  #refine foreground is an extract postprocessing step that increases inference time but can improve matting edges. The default value is False.