Update requirements.txt

requirements.txt

@@ -1,32 +1,29 @@
-# Core ML frameworks - Standard versions for ZeroGPU compatibility
-torch==2.1.0
-torchvision==0.16.0
-
-# Transformers for Florence-2 model
-transformers==4.44.0
-
-# Florence-2 dependencies
-einops==0.8.0
-timm==1.0.3
-
-# Gradio UI framework  
+# BAGEL 7B dependencies - Exact from working implementation
+decord==0.6.0
+einops==0.8.1
+huggingface_hub==0.29.1
+matplotlib==3.7.0
+numpy==1.24.4
+opencv_python==4.7.0.72
+pyarrow==11.0.0
+PyYAML==6.0.2
+requests==2.32.3
+safetensors==0.4.5
+scipy==1.10.1
+sentencepiece==0.1.99
+torch==2.5.1
+torchvision==0.20.1
+transformers==4.49.0
+accelerate>=0.34.0
+wandb
+
+# Gradio UI framework
 gradio==5.33.2
 
-# Image processing
-pillow==10.0.1
-
-# Essential utilities
-numpy==1.24.3
-requests==2.31.0
-
 # Hugging Face Spaces support
-spaces==0.19.4
+spaces==0.37.0
 
-# Optional accelerations (commented out to avoid conflicts)
-# accelerate==0.24.1
-# sentence-transformers==2.2.2
-# flash_attn  # Optional, can work without it
+# Image processing
+pillow==10.0.1
 
-# Development and debugging (optional)
-# logging
-# typing-extensions
+# Flash attention will be installed dynamically in app.py