YANGYYYY commited on
Commit
df4d2e0
1 Parent(s): d45b300

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -233,7 +233,7 @@ class Predictor:
233
  video_writer.close()
234
  #output_file.close()
235
 
236
- def transform_video(self, video_frames, batch_size=4):
237
  #end = end or None
238
 
239
  # if not os.path.isfile(input_path):
 
233
  video_writer.close()
234
  #output_file.close()
235
 
236
+ def transform_video(self, video_frames, batch_size=4,start,end):
237
  #end = end or None
238
 
239
  # if not os.path.isfile(input_path):