KeivanR commited on
Commit
75a89f3
·
1 Parent(s): f86da66
Files changed (1) hide show
  1. Dockerfile +3 -5
Dockerfile CHANGED
@@ -30,11 +30,9 @@ RUN pip install --no-cache-dir \
30
  COPY . .
31
 
32
  # Special model loading step
33
- RUN python3 -c "
34
- from model import QwenClassifier
35
- QwenClassifier.from_pretrained('KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446')
36
- print('Model loaded successfully')
37
- "
38
 
39
  # Run FastAPI app
40
  EXPOSE 7860
 
30
  COPY . .
31
 
32
  # Special model loading step
33
+ RUN python3 -c "from model import QwenClassifier; \
34
+ QwenClassifier.from_pretrained('KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446'); \
35
+ print('Model loaded successfully')"
 
 
36
 
37
  # Run FastAPI app
38
  EXPOSE 7860