Add flask

Author: chaodreaming

app.py

@@ -0,0 +1,39 @@
+import json
+from flask import Flask
+from flask import request
+
+
+from simplelatexocr.models import Latex_OCR
+REQUEST_ID_HEADER = 'x-fc-request-id'
+
+detect_path="models/best.onnx"
+encoder_path = "models/encoder.onnx"
+decoder_path = "models/decoder.onnx"
+tokenizer_json = "models/tokenizer.json"
+
+model = Latex_OCR(
+        detect_path=detect_path,
+        encoder_path=encoder_path,
+        decoder_path=decoder_path,
+        tokenizer_json=tokenizer_json,
+    )
+
+app = Flask(__name__)
+
+@app.route('/', methods=['GET', 'POST'])
+def upload_image():
+    if request.method == 'POST' and 'image' in request.files:
+        image = request.files['image']
+        res = model.predict(image)
+        print("formula: ", res["formula"])
+        print("confidence: ", res["confidence"])
+        print("elapse: ", res["elapse"])
+        print(res)
+        return json.dumps(res)
+    else:
+        return json.dumps({"error": "info erro"})
+
+if __name__ == '__main__':
+
+    app.run(host='0.0.0.0',port=9000,debug=False)
+

requirements.txt

@@ -4,4 +4,5 @@ numpy
 opencv-python
 Pillow>=9.2.0
 PyYAML
-transformers
+transformers
+flask

test.py

@@ -0,0 +1,15 @@
+import json
+import requests
+
+
+image_path = "tests/test_files/5.png"
+url = "http://127.0.0.1:9000/"
+files = {'image': open(image_path, 'rb')}
+response = requests.post(url, files=files)
+
+# print result
+print(response.text)
+data=json.loads(response.text)
+print("formula:\n{}".format(data['formula']))
+print('confidence:\n{}'.format(data['confidence']))
+print('elapse:\n{}'.format(data['elapse']))