Update app.py

app.py

@@ -32,8 +32,8 @@ def process_image(image, task_token):
     generated_ids = model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],
-        max_new_tokens=1024,
-        num_beams=3,
+        max_new_tokens=256,
+        num_beams=2,
         do_sample=False
     )
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
@@ -41,7 +41,7 @@ def process_image(image, task_token):
     return parsed_result
 
 @app.post("/process-image/")
-async def process_image_endpoint(file: UploadFile = File(...), task_token: str = "<OD>"):
+async def process_image_endpoint(file: UploadFile = File(...), task_token: str = Form(" ")):
     image = Image.open(io.BytesIO(await file.read())).convert("RGB")
     result = process_image(image, task_token)
     return result