Upload 2 files

Dockerfile

@@ -4,12 +4,17 @@ WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Upgrade pip before installing requirements
+RUN pip install --no-cache-dir --upgrade pip
 
-RUN useradd user
+# Install dependencies
+RUN pip install --no-cache-dir -r /code/requirements.txt
 
+# Create and use non-root user
+RUN useradd -m user
 USER user
 
+# Set environment variables
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:${PATH}
 
@@ -17,4 +22,5 @@ WORKDIR $HOME/app
 
 COPY --chown=user . $HOME/app
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -1,11 +1,10 @@
-# Adjust FastAPI and mistral-common to compatible versions
-fastapi==0.73.0  # Downgrade FastAPI to match pydantic version required by mistral-common
-requests==2.27.*
-uvicorn[standard]==0.17.*
-sentencepiece==0.2.0  # Use version required by mistral-common
+fastapi==0.95.0
+requests==2.28.2
+uvicorn[standard]==0.20.0
+sentencepiece==0.2.0
 torch>=1.13.0
 transformers==4.28.0
 numpy<2
 peft==0.11.1
-huggingface-hub==0.17.0
-mistral-common==1.3.0  # Version compatible with FastAPI and pydantic
+huggingface-hub>=0.17.0
+mistral-common==1.3.0