Update Dockerfile

Dockerfile

@@ -6,14 +6,15 @@ COPY requirements.txt .
 
 RUN apt-get update && \
     apt-get install -y --no-install-recommends gcc python3-dev && \
-    pip install --user -r requirements.txt
+    pip install --user -r requirements.txt gunicorn && \
+    rm -rf /var/lib/apt/lists/*
 
 # Stage 2: Runtime
 FROM python:3.9-slim
 
 WORKDIR /app
 
-# Copy only necessary files from builder
+# Copy Python dependencies from builder
 COPY --from=builder /root/.cache /root/.cache
 COPY --from=builder /usr/local/lib/python3.9/site-packages /usr/local/lib/python3.9/site-packages
 
@@ -25,9 +26,12 @@ RUN apt-get update && \
 # Copy application code
 COPY app.py text2generation.py ./
 
-# Configure production server
+# Expose Hugging Face's default port
 EXPOSE 7860
-HEALTHCHECK --interval=30s --timeout=30s --start-period=120s --retries=3 \
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=30s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
+# Run with Gunicorn (now properly installed)
 CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--timeout", "120", "app:app"]