add docker config

Dockerfile

@@ -0,0 +1,30 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+# system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY main.py .
+COPY inference_utils.py .
+
+RUN mkdir -p packaged_models
+
+COPY ./packaged_models/model.scripted.pt ./packaged_models/
+COPY ./packaged_models/model.pt ./packaged_models/
+
+# Expose the port
+EXPOSE 7860
+
+# Set environment variables
+ENV PYTHONPATH=/app
+ENV PYTHONUNBUFFERED=1
+
+# Run the application
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -3,7 +3,7 @@ title: Scriptify Api
 emoji: ✍️
 colorFrom: indigo
 colorTo: green
-sdk: python
+sdk: docker
 app_file: main.py 
 python_version: 3.9
 pinned: false

main.py

@@ -13,9 +13,9 @@ from inference_utils import construct_alphabet_list, convert_offsets_to_absolute
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__) 
 
-MODEL_DIR = Path("../../ml/packaged_models") 
-SCRIPTED_MODEL_NAME = "handwriting_model.scripted.pt" 
-METADATA_MODEL_NAME = "handwriting_model.pt" 
+MODEL_DIR = Path("./packaged_models") 
+SCRIPTED_MODEL_NAME = "model.scripted.pt" 
+METADATA_MODEL_NAME = "model.pt" 
 
 scripted_model: Optional[torch.jit.ScriptModule] = None 
 model_metadata: Optional[dict] = None