Update Dockerfile

Dockerfile

@@ -1,42 +1,35 @@
-FROM ubuntu:22.04
+# Use official Python 3.10 slim image
+FROM python:3.10-slim
 
-# Set time zone to Asia/Kolkata (IST)
-ENV TZ=Asia/Kolkata
-RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
-
-# Set cache directories for Hugging Face and Numba
-ENV HF_HOME=/app/cache
-ENV NUMBA_CACHE_DIR=/app/cache
-ENV VLLM_CPU_ONLY=True
-RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+# Set working directory
+WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-    python3.10 \
-    python3-pip \
     tesseract-ocr \
     libtesseract-dev \
     poppler-utils \
-    libopencv-dev \
-    tzdata \
-    && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# Pre-install NumPy to enforce version
-RUN pip3 install --no-cache-dir numpy==1.26.4
-
-# Pre-install torch to avoid vLLM build issues
-RUN pip3 install --no-cache-dir torch==2.4.0
+# Copy requirements file
+COPY requirements.txt .
 
 # Install Python dependencies
-COPY requirements.txt .
-RUN pip3 install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
-COPY app.py .
+COPY expense.py .
+COPY app.py .  # Optional: Include if you want to keep /ocr endpoint
+
+# Create cache directory
+RUN mkdir -p /app/cache
+
+# Set environment variables
+ENV HF_HOME=/app/cache
+ENV NUMBA_CACHE_DIR=/app/cache
 
-# Expose port for FastAPI (Hugging Face Spaces default)
+# Expose port
 EXPOSE 7860
 
-# Command to run FastAPI
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run FastAPI with expense.py (default: /categorize endpoint)
+CMD ["uvicorn", "expense:app", "--host", "0.0.0.0", "--port", "7860"]