Update Dockerfile

Dockerfile

@@ -1,17 +1,29 @@
-# Base image
-FROM python:3.10-slim
+# Use lightweight Python image
+FROM python:3.9-slim
+
+# Install system dependencies for transformers
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends gcc python3-dev && \
+    rm -rf /var/lib/apt/lists/*
 
 # Set working directory
 WORKDIR /app
 
-# Copy all files
-COPY . /app
-
-# Install dependencies
+# Install Python dependencies
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose the port you plan to use (match app.run port)
-EXPOSE 8080
+# Cache the AI model (Falcon-7B)
+RUN python -c """
+from transformers import pipeline
+pipeline('text-generation', model='tiiuae/falcon-7b-instruct')
+"""
+
+# Copy application files
+COPY app.py text2generation.py ./
+
+# Expose Hugging Face's default port
+EXPOSE 7860
 
-# Run the Flask app
-CMD ["python", "app.py"]
+# Run the application with Gunicorn for production
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "2", "app:app"]