Spaces:
Sleeping
Sleeping
John Ho
commited on
Commit
·
c74b32a
1
Parent(s):
a32aeaf
forgot to load model
Browse files
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 = [
|