minor

Dockerfile

@@ -18,6 +18,9 @@ ENV PATH="/opt/venv/bin:$PATH"
 
 # Install Python dependencies first (for better caching)
 COPY requirements.txt .
+COPY app.py .
+COPY qwen_classifier/ ./qwen_classifier/
+COPY setup.py . 
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Install PyTorch with CUDA support
@@ -26,15 +29,9 @@ RUN pip install --no-cache-dir \
     torchvision==0.16.2+cu121 \
     --extra-index-url https://download.pytorch.org/whl/cu121
 
-# Copy entire application (including model.py)
-COPY . .
-
-# Install your package in development mode
-RUN pip install -e .
-
 # Set up Hugging Face authentication (use a build ARG for the token)
-ARG HF_TOKEN
-RUN python3 -c "from huggingface_hub import login; login(token='$HF_TOKEN')"
+# ARG HF_TOKEN
+# RUN python3 -c "from huggingface_hub import login; login(token='$HF_TOKEN')"
 
 # Test model loading (use absolute import path)
 RUN python3 -c "from qwen_classifier.model import QwenClassifier; \