From cda479527015217e813252f532c3bd81f6c5d5b4 Mon Sep 17 00:00:00 2001 From: Thomas Dhome-Casanova Date: Mon, 27 Jan 2025 22:40:37 -0800 Subject: [PATCH] add omniparse is ready endpoint --- demo/remote_request.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/demo/remote_request.py b/demo/remote_request.py index 95f3595..84dd3a9 100644 --- a/demo/remote_request.py +++ b/demo/remote_request.py @@ -3,14 +3,15 @@ import sys import os sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) - +import time from utils import get_som_labeled_img, get_caption_model_processor, get_yolo_model, check_ocr_box import torch from PIL import Image from typing import Dict, Tuple, List import base64 import io - +from fastapi import FastAPI +from pydantic import BaseModel config = { 'som_model_path': '../weights/icon_detect_v1_5/model_v1_5.pt', @@ -51,8 +52,7 @@ class Omniparser(object): return dino_labled_img, parsed_content_list -from fastapi import FastAPI -from pydantic import BaseModel + app = FastAPI() @@ -65,9 +65,13 @@ Omniparser = Omniparser(config) @app.post("/send_text/") async def send_text(item: Item): print('start parsing...') - import time + start = time.time() dino_labled_img, parsed_content_list = Omniparser.parse(item.base64_image) latency = time.time() - start print('time:', latency) - return {"som_image_base64": dino_labled_img, "parsed_content_list": parsed_content_list, 'latency': latency} \ No newline at end of file + return {"som_image_base64": dino_labled_img, "parsed_content_list": parsed_content_list, 'latency': latency} + +@app.get("/") +async def root(): + return {"message": "Omniparser API ready"} \ No newline at end of file