updated requirements.txt and dockerfile

Dockerfile

@@ -1,32 +1,24 @@
-# ✅ Use official slim image
 FROM python:3.10-slim
 
-# ✅ Set working directory
 WORKDIR /app
 
-# ✅ Set environment variables early to ensure cache use in setup.py
 ENV TRANSFORMERS_CACHE=/app/model_cache \
     HF_HOME=/app/model_cache \
     TORCH_HOME=/app/model_cache \
     TOKENIZERS_PARALLELISM=false \
     OMP_NUM_THREADS=4
 
-# ✅ Install only necessary OS packages and clean cache
 RUN apt-get update && apt-get install -y git \
  && rm -rf /var/lib/apt/lists/*
 
-# ✅ Copy files (excluding model_cache, logs, etc. via .dockerignore)
 COPY . .
 
-# ✅ Upgrade pip + install deps without cache
 RUN pip install --upgrade pip \
  && pip install --no-cache-dir -r requirements.txt
 
-# ✅ Run setup.py to download the model
-RUN python setup.py
+# Optional: preload model
+RUN python download_model.py
 
-# ✅ Expose Hugging Face Space-required port
 EXPOSE 7860
 
-# ✅ Launch FastAPI on port 7860 for HF Space
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--timeout-keep-alive", "120", "--log-level", "info"]

requirements.txt

@@ -1,24 +1,17 @@
-# torch>=2.1.0
-# transformers>=4.40.0
-# accelerate>=0.25.0
-# bitsandbytes
-# fastapi
-# uvicorn
-#Your original dependencies (optimized versions)
+# Core dependencies
 torch>=2.1.0
 transformers==4.41.1
 accelerate==0.30.1
 bitsandbytes
+tokenizers>=0.15.0
+safetensors>=0.4.0
+huggingface-hub>=0.19.0
+
+# FastAPI web backend
 fastapi
 uvicorn[standard]
+python-multipart  # For handling file uploads in FastAPI
+python-dotenv     # For loading environment variables
 
-# Additional optimizations for faster loading
-tokenizers>=0.15.0  # Fast tokenizers (auto-installed with transformers but explicit for optimization)
-safetensors>=0.4.0  # Faster model loading format
-huggingface-hub>=0.19.0  # Better caching and download management
-
-# Optional performance improvements
-psutil>=5.9.0  # For system monitoring
-python-multipart  # For FastAPI file uploads if needed
-
-python-dotenv
+# Optional: System monitoring / logging (optional but useful)
+psutil>=5.9.0