Spaces:

Frederic-CellNum
/

api_sparrow_ocr

Sleeping

Frederic-CellNum commited on about 1 month ago

Commit

941ccd9

verified ·

1 Parent(s): 82aed82

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -7,9 +7,8 @@ COPY requirements.txt .
 RUN pip install -r requirements.txt
 # Pré-télécharger le modèle PENDANT le build (accès réseau autorisé)
-RUN python -c "from transformers import Qwen2VLForConditionalGeneration, AutoProcessor"
-RUN python -c "AutoProcessor.from_pretrained('Qwen/Qwen2-VL-7B-Instruct')"
-RUN python -c "Qwen2VLForConditionalGeneration.from_pretrained('Qwen/Qwen2-VL-7B-Instruct', torch_dtype='auto')"
 COPY . .

 RUN pip install -r requirements.txt
 # Pré-télécharger le modèle PENDANT le build (accès réseau autorisé)
+COPY download_model.py .
+RUN python download_model.py
 COPY . .