统一模型接口,新增文本分类接口

This commit is contained in:
2024-09-27 13:50:55 +08:00
parent 117b29a737
commit f1149854ce
13 changed files with 144 additions and 97 deletions

View File

@@ -0,0 +1,22 @@
import logging.config
from flask import Flask, request
from paddleocr import PaddleOCR
from log import LOGGING_CONFIG
from utils import process_request
app = Flask(__name__)
OCR = PaddleOCR(use_angle_cls=False, show_log=False, gpu_id=0, det_db_box_thresh=0.3)
@app.route('/', methods=['POST'])
@process_request
def main():
img_path = request.form.get('img_path')
return OCR.ocr(img_path, cls=False)
if __name__ == '__main__':
logging.config.dictConfig(LOGGING_CONFIG)
app.run('0.0.0.0', 5001)