fix: add pip upgrade

Dockerfile

@@ -31,7 +31,7 @@ ENV HOME=/home/user \
     PYTHONUNBUFFERED=1 \
     GRADIO_SERVER_NAME=0.0.0.0
 
-RUN pip3 install --use-deprecated=legacy-resolver --no-cache-dir --upgrade -r /code/requirements.txt
+RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app

backends/smoldocling.py

@@ -18,7 +18,6 @@ processor = AutoProcessor.from_pretrained("ds4sd/SmolDocling-256M-preview")
 model = AutoModelForVision2Seq.from_pretrained(
     "ds4sd/SmolDocling-256M-preview",
     torch_dtype=torch.bfloat16,
-    _attn_implementation="flash_attention_2" if DEVICE == "cuda" else "eager",
 ).to(DEVICE)
 
 # Create input messages

requirements.txt

@@ -10,4 +10,3 @@ transformers<5.0.0,>=4.45.2
 pypdf
 docling_core
 opencv-contrib-python
-flash_attn