docker and run.sh

Dockerfile

@@ -1,10 +1,25 @@
+# Use official lightweight Python image
 FROM python:3.10-slim
 
+# Set environment variables
+ENV HF_HOME=/data
+ENV PYTHONUNBUFFERED=1
+
+# Create Hugging Face hub directory
+RUN mkdir -p /data/hub
+
+# Set working directory
 WORKDIR /app
 
+# Install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy code
 COPY . .
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Ensure run.sh is executable
+RUN chmod +x run.sh
+
+# Default command
+CMD ["./run.sh"]

README.md

@@ -3,10 +3,6 @@ title: SYPLYD MARBERT API
 emoji: 🌍
 colorFrom: green
 colorTo: blue
-sdk: fastapi
-app_file: app.py
+sdk: docker
+app_file: run.sh
 ---
-
-# SYPLYD MARBERT API Endpoint
-
-This Space serves the `ShazaAly/syplyd-marbert-1` model via a FastAPI API.

run.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# Set the Hugging Face home directory to the persistent /data folder
+export HF_HOME=/data
+
+# Create the directory that the transformers library will try to use.
+# The -p flag ensures it doesn't fail if the directory already exists.
+mkdir -p $HF_HOME/hub
+
+# Now, start the FastAPI application
+# Using exec means this script will be replaced by the uvicorn process, which is efficient.
+exec uvicorn app:app --host 0.0.0.0 --port 7860