Update Dockerfile

Dockerfile

@@ -1,25 +1,21 @@
-# Use official Python image
-FROM python:3.9
+# Use Ubuntu 22.04 for updated system libraries
+FROM ubuntu:22.04
 
-# ✅ Set cache directory for Hugging Face
-ENV HF_HOME=/home/user/cache
+# Set non-interactive mode (prevents prompts)
+ENV DEBIAN_FRONTEND=noninteractive
 
-# ✅ Create cache directories & set correct permissions before switching users
-RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
+# Install system dependencies (Fixes GLIBCXX issue)
+RUN apt update && apt install -y libstdc++6 python3 python3-pip
 
-# ✅ Install dependencies
-RUN pip install --no-cache-dir --upgrade fastapi uvicorn transformers torch bitsandbytes accelerate
-
-# ✅ Create a non-root user (for security) and switch to it
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-
-# Set working directory
+# Set working directory inside the container
 WORKDIR /app
 
+# Copy requirements.txt and install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
 # Copy application files
 COPY app.py .
 
-# Run the FastAPI server
+# Run the FastAPI app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]