Update Dockerfile

Dockerfile

@@ -1,26 +1,26 @@
-# Use a lightweight Python base image
+# Use official Python 3.10 slim image for a smaller footprint
 FROM python:3.10-slim
 
 # Set working directory
 WORKDIR /app
 
-# Copy requirements.txt
+# Copy requirements file
 COPY requirements.txt .
 
-# Install dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+# Install system dependencies and Python packages
+RUN apt-get update && apt-get install -y \
+    gcc \
+    && rm -rf /var/lib/apt/lists/* \
+    && pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
-COPY . .
+COPY app.py .
 
-# Set environment variables for Hugging Face cache
-ENV HF_HOME=/app/cache
-ENV TRANSFORMERS_CACHE=/app/cache
-# Set to 1 if using quantization and GPU is available
-ENV USE_QUANTIZATION=0
+# Create cache directory for Hugging Face models
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 
 # Expose port for FastAPI
 EXPOSE 8000
 
-# Run the FastAPI app with uvicorn (updated to use app.py)
+# Run Uvicorn server
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]