Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
+# Use Python 3.9 base image
 FROM python:3.9-slim
 
-WORKDIR /app
-
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -9,13 +9,20 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Set working directory
+WORKDIR /app
+
+# Copy all files
+COPY . .
 
-RUN pip3 install -r requirements.txt
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
-EXPOSE 8501
+# Download model during build
+RUN python -c "from transformers import AutoTokenizer, AutoModelForSeq2SeqLM; AutoTokenizer.from_pretrained('tscholak/cxmefzzi'); AutoModelForSeq2SeqLM.from_pretrained('tscholak/cxmefzzi')"
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Expose port
+EXPOSE 7860
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Start Streamlit
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]