diff --git a/src/webserver.py b/src/webserver.py index e1c19e4c384d692519f43e67c643cbaeb6d41b6e..52dc71d007c94ab8a6ff0db5ef0633882842bc61 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)