Update Dockerfile

Dockerfile

@@ -10,24 +10,11 @@ COPY requirements.txt /code/requirements.txt
 # Install any dependencies specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Apply the patch to remove flash-attn dependency and download the Florence-2-base-ft model and processor
-RUN python -c "\
-import os; \
-from unittest.mock import patch; \
-from transformers import AutoModelForCausalLM, AutoProcessor; \
-from transformers.dynamic_module_utils import get_imports; \
-def fixed_get_imports(filename: os.PathLike) -> list[str]: \
-    if not str(filename).endswith('/modeling_florence2.py'): \
-        return get_imports(filename); \
-    imports = get_imports(filename); \
-    if 'flash_attn' in imports: \
-        imports.remove('flash_attn'); \
-    return imports; \
-with patch('transformers.dynamic_module_utils.get_imports', fixed_get_imports): \
-    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 Python script to download the model
+COPY download_model.py /code/download_model.py
+
+# Run the Python script to download the Florence-2-base-ft model and processor
+RUN python /code/download_model.py
 
 # Copy the current directory contents into the container at /code
 COPY . .