Use Docker SDK with model download from HF Hub

.gitignore

@@ -0,0 +1,25 @@
+# Model files (downloaded at runtime)
+animal-classification/
+model_cache/
+
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+env/
+venv/
+ENV/
+.venv
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+Thumbs.db

Dockerfile

@@ -0,0 +1,24 @@
+FROM python:3.12-slim
+
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements and install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code (model will be downloaded at runtime)
+COPY app.py .
+
+# Expose port 7860 (Hugging Face Spaces default)
+EXPOSE 7860
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+
+# Run the application on port 7860
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -3,9 +3,8 @@ title: Animal Classification API
 emoji: 🐾
 colorFrom: blue
 colorTo: green
-sdk: fastapi
-sdk_version: "0.115.4"
-app_file: app.py
+sdk: docker
+app_port: 7860
 pinned: false
 ---