Update Dockerfile

Dockerfile

@@ -1,18 +1,15 @@
-FROM nvidia/cuda:12.1-base-ubuntu22.04
+FROM python:3.11-slim
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV DEBIAN_FRONTEND=noninteractive
 ENV HF_HOME=/app/huggingface
 ENV TRANSFORMERS_CACHE=/app/huggingface
+ENV PORT=7860
+ENV HOST=0.0.0.0
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-    python3 \
-    python3-pip \
-    python3-venv \
-    git \
-    wget \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
@@ -21,7 +18,7 @@ WORKDIR /app
 
 # Copy requirements and 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 . .