Update Dockerfile

Dockerfile

@@ -3,16 +3,16 @@ FROM python:3.9-slim
 
 # Set the working directory in the container
 WORKDIR /code
-RUN mkdir -p /code/.cache
 
-ENV TRANSFORMERS_CACHE=/code/.cache
 # Copy the requirements.txt file into the container at /code
 COPY requirements.txt /code/requirements.txt
 
-
 # Install any dependencies specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Download the Florence-2-base-ft model and processor
+RUN python -c "from transformers import AutoModelForCausalLM, AutoProcessor; model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base-ft', trust_remote_code=True); processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base-ft', trust_remote_code=True); model.save_pretrained('/code/florence_model'); processor.save_pretrained('/code/florence_processor')"
+
 # Copy the current directory contents into the container at /code
 COPY . .