Upload folder using huggingface_hub

inference.py

@@ -30,6 +30,13 @@ API_KEY = os.environ.get("API_KEY", "")
 MAX_WAIT_SECONDS = 900  # 15 minutes
 POLL_INTERVAL_SECONDS = 5
 
+# Default processing parameters (LearningStudio doesn't pass these)
+DEFAULT_PARAMS = {
+    "tiling": {"tile": 2048, "overlap": 0.30},
+    "floodfill": {"erase_text": False, "min_fill_vs_text": 0.0},
+    "preclean": {"denoise_sw": 8}
+}
+
 
 def normalize_to_bytes(image_input: str) -> Tuple[bytes, str]:
     """
@@ -308,9 +315,10 @@ def inference(image_input: str, parameters: Optional[Dict] = None) -> Dict[str,
         logger.info("Uploading to S3...")
         upload_to_s3(upload_url, image_bytes)
 
-        # 4. Start detection job
+        # 4. Start detection job (merge user params with defaults)
         logger.info("Starting detection job...")
-        start_detection_job(job_id, s3_url, parameters)
+        merged_params = {**DEFAULT_PARAMS, **(parameters or {})}
+        start_detection_job(job_id, s3_url, merged_params)
 
         # 5. Poll for completion
         logger.info("Polling for completion...")

test_inference.sh

@@ -0,0 +1,46 @@
+#!/bin/bash
+
+# Test the HuggingFace Inference Endpoint with a local image file
+# Usage: ./test_inference.sh <image_file>
+
+set -e
+
+if [ -z "$1" ]; then
+    echo "Usage: $0 <image_file>"
+    exit 1
+fi
+
+IMAGE_FILE="$1"
+
+if [ ! -f "$IMAGE_FILE" ]; then
+    echo "Error: File not found: $IMAGE_FILE"
+    exit 1
+fi
+
+# Required environment variables
+if [ -z "$HF_ENDPOINT" ]; then
+    echo "Error: HF_ENDPOINT not set"
+    echo "Export it first: export HF_ENDPOINT=https://xxx.endpoints.huggingface.cloud"
+    exit 1
+fi
+
+if [ -z "$HF_TOKEN" ]; then
+    echo "Error: HF_TOKEN not set"
+    echo "Export it first: export HF_TOKEN=hf_xxx"
+    exit 1
+fi
+
+# Convert image to base64 and create JSON payload in temp file
+echo "Converting $IMAGE_FILE to base64..."
+TEMP_FILE=$(mktemp)
+trap "rm -f $TEMP_FILE" EXIT
+
+printf '{"inputs": "' > "$TEMP_FILE"
+base64 -i "$IMAGE_FILE" | tr -d '\n' >> "$TEMP_FILE"
+printf '"}' >> "$TEMP_FILE"
+
+echo "Sending request to $HF_ENDPOINT..."
+curl -s -X POST "$HF_ENDPOINT" \
+    -H "Authorization: Bearer $HF_TOKEN" \
+    -H "Content-Type: application/json" \
+    -d @"$TEMP_FILE" | python3 -m json.tool