Pre-install flag added in README.md

README.md

@@ -9,6 +9,7 @@ app_file: app.py
 python_version: "3.10"
 pinned: false
 license: apache-2.0
+preinstall: "bash pre-install.sh"
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -37,7 +37,7 @@ print("Model loaded.")
 VIDEO_IDS = sorted([
     os.path.splitext(f)[0]
     for f in os.listdir(VIDEO_DIR)
-    if f.endswith(".webm")
+    if f.endswith(".mp4")
 ])
 
 # ----------------------
@@ -46,7 +46,7 @@ VIDEO_IDS = sorted([
 def get_video_path(video_id: str):
     if not video_id:
         return None
-    path = os.path.join(VIDEO_DIR, video_id + ".webm")
+    path = os.path.join(VIDEO_DIR, video_id + ".mp4")
     return path if os.path.exists(path) else None
 
 # ----------------------

models/llava_video.py

@@ -56,6 +56,7 @@ class LLaVAVideoModel(BaseVideoModel):
                 base_model,
                 torch_dtype=torch_dtype,
                 device_map=device_map,
+                attn_implementation=attn_implementation,
             )
         )  # Add any other thing you want to pass in llava_model_args
         self.model.eval()

pre-install.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+# Pre-install script for HF Spaces
+# This installs llava without dependencies to avoid version conflicts
+
+echo "Installing LLaVA-NeXT without dependencies..."
+pip install --no-deps git+https://github.com/LLaVA-VL/LLaVA-NeXT.git
+
+echo "LLaVA-NeXT installed successfully without dependency resolution"
+

requirements.txt

@@ -15,7 +15,8 @@ h11==0.16.0
 hf-xet==1.2.0
 hf_transfer==0.1.8
 huggingface-hub>=0.21.0,<1.0
-llava @ git+https://github.com/LLaVA-VL/LLaVA-NeXT.git#egg=llava[standalone] --no-deps
+# llava @ git+https://github.com/LLaVA-VL/LLaVA-NeXT.git#egg=llava
+# NOTE: LLaVA is installed via pre-install.sh without dependencies to avoid conflicts
 matplotlib==3.10.8
 matplotlib-inline==0.2.1
 mpmath==1.3.0