Update app.py

app.py

@@ -10,7 +10,7 @@ from PIL import Image
 from huggingface_hub import hf_hub_download
 import io
 
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.responses import JSONResponse
 import uvicorn
 
@@ -94,14 +94,18 @@ def image_to_deepdanbooru_tags(
 
     return output_text, filtered_tags
 
+from typing import Optional
+
 @app.post("/tagging")
-async def tagging_endpoint(image: UploadFile = File(...)):
+async def tagging_endpoint(
+    image: UploadFile = File(...),
+    threshold: Optional[float] = Form(0.5) 
+):
     image_data = await image.read()
     pil_image = Image.open(io.BytesIO(image_data)).convert("RGB")
-    # Use default parameters or customize as needed
     output_text, filtered_tags = image_to_deepdanbooru_tags(
         pil_image,
-        threshold=0.5,
+        threshold=threshold,
         use_spaces=False,
         use_escape=False,
         include_ranks=False,