dockerfile

Dockerfile

@@ -29,8 +29,15 @@ RUN pip install --no-cache-dir \
 # Copy entire application (including model.py)
 COPY . .
 
-# Special model loading step
-RUN python3 -c "from model import QwenClassifier; \
+# 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')"
+
+# Test model loading (use absolute import path)
+RUN python3 -c "from qwen_classifier.model import QwenClassifier; \
     QwenClassifier.from_pretrained('KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446'); \
     print('Model loaded successfully')"