Upload 25 files

Dockerfile

@@ -13,8 +13,8 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
 
 # Install basic dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3.9 \
-    python3.9-dev \
+    python3.11 \
+    python3.11-dev \
     python3-pip \
     python3-setuptools \
     git \
@@ -26,10 +26,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
 
 # Create symbolic link for python
-RUN ln -sf /usr/bin/python3.9 /usr/bin/python
+RUN ln -sf /usr/bin/python3.11 /usr/bin/python
 
 # Upgrade pip
-RUN python3.9 -m pip install --no-cache-dir --upgrade pip
+RUN python3.11 -m pip install --no-cache-dir --upgrade pip
 
 # Create cache directories
 RUN mkdir -p /app/.cache/transformers \
@@ -37,23 +37,13 @@ RUN mkdir -p /app/.cache/transformers \
     /app/.cache/torch \
     /app/.cache/datasets
 
-# Install PyTorch with CUDA support
-RUN pip install --no-cache-dir torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --extra-index-url https://download.pytorch.org/whl/cu118
-
-# Install latest version of transformers that includes Cache class
-RUN pip install --no-cache-dir transformers>=4.34.0
-
-# Clone and install LAVIS properly
-RUN git clone https://github.com/salesforce/LAVIS.git \
-    && cd LAVIS \
-    && sed -i '/open3d/d' requirements.txt \
-    && sed -i 's/transformers.*//' requirements.txt \
-    && python3.9 -m pip install --no-cache-dir -e .
+RUN cd LAVIS \
+    && python3.11 -m pip install --no-cache-dir -e .
 
 # Copy requirements files and install dependencies
 # Note: We'll install without replacing transformers
-COPY requirements_lavis.txt requirements_emo.txt ./
-RUN pip install --no-cache-dir --no-deps -r requirements_lavis.txt -r requirements_emo.txt
+COPY requirements.txt .
+RUN pip install --no-cache-dir --no-deps -r requirements.txt
 
 # Copy application files
 COPY app.py ./

blip2_vicuna_instruct.py

@@ -49,7 +49,7 @@ class Blip2VicunaInstruct(Blip2Base):
     ):
         super().__init__()
         transformers_version = version.parse(transformers.__version__)
-        assert transformers_version >= version.parse("4.28"), "BLIP-2 Vicuna requires transformers>=4.28"        
+        # assert transformers_version >= version.parse("4.28"), "BLIP-2 Vicuna requires transformers>=4.28"        
         from transformers import LlamaTokenizer
         from lavis.models.blip2_models.modeling_llama import LlamaForCausalLM
         

emo/train.json

@@ -1,3 +1,3 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b3043ba41320d9309513d9661e64c1f89d2c40370302596718b774f5917e772f
-size 10750390
+version https://git-lfs.github.com/spec/v1
+oid sha256:b3043ba41320d9309513d9661e64c1f89d2c40370302596718b774f5917e772f
+size 10750390

start.sh

@@ -1,77 +1,77 @@
-#!/bin/bash
-set -e
-
-echo "🚀 Starting EmoVIT application..."
-
-# Create and set permissions for cache directories
-echo "📁 Setting up cache directories..."
-mkdir -p /app/.cache/huggingface/hub
-mkdir -p /app/.cache/transformers
-mkdir -p /app/.cache/torch
-mkdir -p /app/.cache/datasets
-
-# Set permissions only for /app/.cache (avoid root filesystem)
-chmod -R 777 /app/.cache/ 2>/dev/null || true
-
-# Set environment variables
-export TRANSFORMERS_CACHE=${TRANSFORMERS_CACHE:-/app/.cache/transformers}
-export HF_HOME=${HF_HOME:-/app/.cache/huggingface}
-export TORCH_HOME=${TORCH_HOME:-/app/.cache/torch}
-export HF_DATASETS_CACHE=${HF_DATASETS_CACHE:-/app/.cache/datasets}
-export HUGGINGFACE_HUB_CACHE=${HUGGINGFACE_HUB_CACHE:-/app/.cache/huggingface/hub}
-export PYTHONPATH=/app:/app/lib:$PYTHONPATH
-export PYTHONUNBUFFERED=1
-
-echo "✅ Cache directories set up"
-echo "🔧 Environment variables:"
-echo "  TRANSFORMERS_CACHE=$TRANSFORMERS_CACHE"
-echo "  HF_HOME=$HF_HOME" 
-echo "  TORCH_HOME=$TORCH_HOME"
-echo "  PYTHONPATH=$PYTHONPATH"
-
-# Test dependencies
-echo "🔍 Testing dependencies..."
-python3 -c "
-import sys
-import warnings
-warnings.filterwarnings('ignore')
-
-try:
-    import numpy
-    print(f'✅ NumPy {numpy.__version__}')
-except Exception as e:
-    print(f'❌ NumPy: {e}')
-    sys.exit(1)
-
-try:
-    import cv2
-    print(f'✅ OpenCV {cv2.__version__}')
-except Exception as e:
-    print(f'❌ OpenCV: {e}')
-    sys.exit(1)
-
-try:
-    import torch
-    print(f'✅ PyTorch {torch.__version__}')
-except Exception as e:
-    print(f'❌ PyTorch: {e}')
-    sys.exit(1)
-
-try:
-    import transformers
-    print(f'✅ Transformers {transformers.__version__}')
-except Exception as e:
-    print(f'❌ Transformers: {e}')
-    sys.exit(1)
-
-try:
-    import lavis
-    print(f'✅ LAVIS imported successfully')
-except Exception as e:
-    print(f'❌ LAVIS: {e}')
-    print('Warning: LAVIS not available, app will run in demo mode')
-"
-
-echo "🌐 Starting Flask application..."
-# Use unbuffered Python output for better logging in containers
+#!/bin/bash
+set -e
+
+echo "🚀 Starting EmoVIT application..."
+
+# Create and set permissions for cache directories
+echo "📁 Setting up cache directories..."
+mkdir -p /app/.cache/huggingface/hub
+mkdir -p /app/.cache/transformers
+mkdir -p /app/.cache/torch
+mkdir -p /app/.cache/datasets
+
+# Set permissions only for /app/.cache (avoid root filesystem)
+chmod -R 777 /app/.cache/ 2>/dev/null || true
+
+# Set environment variables
+export TRANSFORMERS_CACHE=${TRANSFORMERS_CACHE:-/app/.cache/transformers}
+export HF_HOME=${HF_HOME:-/app/.cache/huggingface}
+export TORCH_HOME=${TORCH_HOME:-/app/.cache/torch}
+export HF_DATASETS_CACHE=${HF_DATASETS_CACHE:-/app/.cache/datasets}
+export HUGGINGFACE_HUB_CACHE=${HUGGINGFACE_HUB_CACHE:-/app/.cache/huggingface/hub}
+export PYTHONPATH=/app:/app/lib:$PYTHONPATH
+export PYTHONUNBUFFERED=1
+
+echo "✅ Cache directories set up"
+echo "🔧 Environment variables:"
+echo "  TRANSFORMERS_CACHE=$TRANSFORMERS_CACHE"
+echo "  HF_HOME=$HF_HOME" 
+echo "  TORCH_HOME=$TORCH_HOME"
+echo "  PYTHONPATH=$PYTHONPATH"
+
+# Test dependencies
+echo "🔍 Testing dependencies..."
+python3 -c "
+import sys
+import warnings
+warnings.filterwarnings('ignore')
+
+try:
+    import numpy
+    print(f'✅ NumPy {numpy.__version__}')
+except Exception as e:
+    print(f'❌ NumPy: {e}')
+    sys.exit(1)
+
+try:
+    import cv2
+    print(f'✅ OpenCV {cv2.__version__}')
+except Exception as e:
+    print(f'❌ OpenCV: {e}')
+    sys.exit(1)
+
+try:
+    import torch
+    print(f'✅ PyTorch {torch.__version__}')
+except Exception as e:
+    print(f'❌ PyTorch: {e}')
+    sys.exit(1)
+
+try:
+    import transformers
+    print(f'✅ Transformers {transformers.__version__}')
+except Exception as e:
+    print(f'❌ Transformers: {e}')
+    sys.exit(1)
+
+try:
+    import lavis
+    print(f'✅ LAVIS imported successfully')
+except Exception as e:
+    print(f'❌ LAVIS: {e}')
+    print('Warning: LAVIS not available, app will run in demo mode')
+"
+
+echo "🌐 Starting Flask application..."
+# Use unbuffered Python output for better logging in containers
 exec python3 -u app.py