API Change

README.md

@@ -23,14 +23,16 @@ A simple REST API around an Ultralytics YOLO model for weld-type detection.
 Set a secret named `API_TOKEN` in your Space (Settings ▸ Secrets).  
 Send requests with `Authorization: Bearer <token>`.
 
-### Request options
+### Python POST (with Requests)
 
-Send `path to image`
+```
+b64_file = base64.b64encode(open("weld-image-file.jpg", "rb").read()).decode()
 
-### cURL examples
+r = requests.post(
+    "https://huggingface.co/spaces/csmith715/ais-api/predict",
+    headers={"Content-Type":"application/json"},
+    json={"image_base64": b64_file}
+)
+print(r.json())
 
-**Multipart file:**
-```bash
-curl -X POST "https://YOUR-SPACE.hf.space/predict" \
-  -H "Authorization: Bearer $API_TOKEN" \
-  -F "file=@sample.jpg"
+```

app.py

@@ -33,7 +33,7 @@ class PredictResponse(BaseModel):
     detections: dict
 
 class PredictQuery(BaseModel):
-    image_base64_path: str
+    image_base64: str
 
 # -----------------------------
 # Utils
@@ -97,8 +97,8 @@ def health():
 
 @app.post("/predict", response_model=PredictResponse)
 def predict(file_data: PredictQuery) -> PredictResponse:
-    file_path = file_data.image_base64_path
-    data = base64.b64encode(open(file_path, "rb").read()).decode()
+    data = file_data.image_base64
+    # data = base64.b64encode(open(file_path, "rb").read()).decode()
     if data is None:
         weld_summary = {}
         # weld_summary = 'No weld types detected.'