Update Dockerfile

Dockerfile

@@ -1,22 +1,21 @@
-# Base image
 FROM python:3.10-slim
 
-# Install system dependencies\ nRUN apt-get update && \
-RUN apt-get update && \
-    apt-get install -y git && \
-    rm -rf /var/lib/apt/lists/*
-
-# Copy and install Python dependencies
-COPY requirements.txt /app/requirements.txt
-RUN pip install --no-cache-dir -r /app/requirements.txt
+# Set environment variables for Transformers cache
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
 
 # Set working directory
 WORKDIR /app
-COPY app.py /app/app.py
 
+# Install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code and model
+COPY . .
 
-# Expose port (Gradio/Streamlit default)
+# Expose Streamlit or FastAPI port (change if you're not using 7860)
 EXPOSE 7860
 
-# Launch
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run your app
+CMD ["python", "app.py"]