Update Dockerfile

Dockerfile

@@ -1,21 +1,33 @@
-FROM python:3.10-slim
+# Use a Python base image
+FROM python:3.9-slim
 
-# Set environment variables for Transformers cache
-ENV TRANSFORMERS_CACHE=/app/cache
-ENV HF_HOME=/app/cache
+# Set environment variables to prevent bytecode generation and buffer output for logging
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
 
-# Set working directory
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set the working directory in the container
 WORKDIR /app
 
-# Install Python dependencies
+# Copy requirements.txt and install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install accelerate
+
+COPY model /app/model
+
 
-# Copy application code and model
+# Copy the application code into the container
 COPY . .
 
-# Expose Streamlit or FastAPI port (change if you're not using 7860)
+# Expose the port your application will run on
 EXPOSE 7860
 
-# Run your app
-CMD ["python", "app.py"]
+# Define the command to run your FastAPI application
+CMD ["python", "app.py"]