John Ho commited on
Commit
c74b32a
·
1 Parent(s): a32aeaf

forgot to load model

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -66,6 +66,7 @@ def inference(
66
  ):
67
  # default processor
68
  processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-7B-Instruct")
 
69
  fps = get_fps_ffmpeg(video_path)
70
  logger.info(f"{os.path.basename(video_path)} FPS: {fps}")
71
  messages = [
 
66
  ):
67
  # default processor
68
  processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-7B-Instruct")
69
+ model = load_model(use_flash_attention=True)
70
  fps = get_fps_ffmpeg(video_path)
71
  logger.info(f"{os.path.basename(video_path)} FPS: {fps}")
72
  messages = [