From c2110a2dfbf8d37dfb609eefff1105122b216ed6 Mon Sep 17 00:00:00 2001
From: merschie <f-mersch@web.de>
Date: Thu, 26 Oct 2023 14:00:32 +0200
Subject: [PATCH] refactoring
---
src/webserver.py | 28 +---------------------------
1 file changed, 1 insertion(+), 27 deletions(-)
diff --git a/src/webserver.py b/src/webserver.py
index e1c19e4..52dc71d 100644
--- a/src/webserver.py
+++ b/src/webserver.py
@@ -43,7 +43,7 @@ def base64_to_image(processed_image):
-@app.route('/predictNachname', methods=['POST'])
+@app.route('/predictName', methods=['POST'])
def predictNach():
image_data=json.loads(request.data)['image']
@@ -69,31 +69,5 @@ def predictNach():
return jsonify(result)
-
-@app.route('/predictVorname', methods=['POST'])
-def predictVor():
- image_data=json.loads(request.data)['image']
-
- image_array=base64_to_image(image_data)
-
- preprocessor = htr_preprocessor.Preprocessor(htr.get_img_size(), dynamic_width=True, padding=16)
- processed_image = preprocessor.process_img(image_array)
- batch = htr_data_loader.Batch([processed_image], None, 1)
-
- #change corpus for name
- names=json.loads(request.data)['names']
- model_name.corpus = names.split(' ')
-
- recognized, probability = htr_model.Model.infer_batch(model_name, batch)
-
- image_base64 = image_to_base64(processed_image)
-
- result = {
- 'recognized': recognized,
- 'image': image_base64
- }
- return jsonify(result)
-
-
if __name__ == '__main__':
app.run(debug=False,port=8000)
--
GitLab