统一模型接口,新增文本分类接口
This commit is contained in:
22
services/paddle_services/ocr.py
Normal file
22
services/paddle_services/ocr.py
Normal 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)
|
||||
Reference in New Issue
Block a user