deploy at 2025-05-13 17:28:26.321602

config.ini

@@ -0,0 +1,6 @@
+[DEFAULT]
+dataset_id = space-backup
+db_dir = data
+private_backup = True
+interval = 15
+

main.py

@@ -4,6 +4,7 @@ from PIL import Image
 from io import BytesIO
 from starlette.responses import JSONResponse
 from fasthtml_hf import setup_hf_backup
+import os
 
 app, rt = fast_app(pico=False, hdrs=(
     Script(src="https://cdn.tailwindcss.com?plugins=forms,typography"),
@@ -211,7 +212,10 @@ def post(filename: str = "black.jpg"):
 @rt('/process')
 def get(filename: str = "black.jpg"):
     # Model inference
-    labrador_learner = load_learner('export.pkl')
+    base_dir = os.path.dirname(os.path.abspath(__file__))
+    model_path = os.path.join(base_dir, 'export.pkl')
+
+    labrador_learner = load_learner(model_path)
     prediction = labrador_learner.predict(filename)
     
     # Extract prediction data