Update app.py

app.py

@@ -1,5 +1,30 @@
 
+import spaces
+import torch
+import argparse
+import os
+import sys
+import pickle  # For serializing frames
+import gc
+import tempfile
+import subprocess
+from datetime import datetime
+from transformers import AutoModel, AutoTokenizer
+from modelscope.hub.snapshot_download import snapshot_download
+from PIL import Image
+from decord import VideoReader, cpu
+import cv2
+import gradio as gr
+from ultralytics import YOLO
+import numpy as np
+import io
 
+# Install flash-attn (using prebuilt wheel mode if needed)
+subprocess.run(
+    'pip install flash-attn --no-build-isolation',
+    env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': 'TRUE'},
+    shell=True
+)
 # --------------------------------------------------------------------
 # Command-line arguments
 # --------------------------------------------------------------------