fix: added

Dockerfile

@@ -1,8 +1,7 @@
-FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
+FROM python:3.11-slim
 
 WORKDIR /app
 
-# libgl1 + libglib2.0-0 are required by Pillow/OpenCV in minimal container environments
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     git \
@@ -10,15 +9,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements and install Python deps
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
 COPY app.py .
 
-# HF Spaces expects port 7860
 EXPOSE 7860
 
-# Run the FastAPI app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -1,6 +1,6 @@
 fastapi==0.115.6
 uvicorn[standard]==0.34.0
-transformers @ git+https://github.com/huggingface/transformers.git@main
-optimum[onnxruntime]
+transformers>=4.51.0
+optimum[onnxruntime]>=1.24.0
 Pillow==11.1.0
 python-multipart==0.0.20