app_v2.py

# -*- coding: utf-8 -*-
"""
Created on Thu Jul 18 11:02:30 2019

@author: Yan
"""
# best ressources are https://scotch.io/bar-talk/processing-incoming-request-data-in-flask
# https://blog.keras.io/building-a-simple-keras-deep-learning-rest-api.html?source=post_page
# https://towardsdatascience.com/deploying-a-keras-deep-learning-model-as-a-web-application-in-p-fc0f2354a7ff


from generate_data import word_to_array
from keras.models import load_model
from flask import Flask, request, jsonify
import tensorflow as tf


app = Flask(__name__)

def load():
    global models
    models = {}
    models['CNN'] = load_model('modelCNN.hdf5')
    models['RNN'] = load_model('modelRNN.hdf5')
    models['FF'] = load_model('modelFF.hdf5')
    global graph
    graph = tf.get_default_graph()
    
    
@app.route('/')

def predict():
        data = {}
        word = request.args.get('word')
        data['word'] = word
        model_name = request.args.get('model')
        data['model'] = model_name
        
        model = models[model_name]
        if model_name == 'FF':
            arr = word_to_array(word).reshape(1,26*12)
        else:
            arr = word_to_array(word).reshape(1,26,12)
            
        with graph.as_default():
            prediction = model.predict(arr)
            data['Francais'] = round(100*prediction[0, 0],2)
            data['English'] = round(100*prediction[0, 1],2)
            data['Espanol'] = round(100*prediction[0, 2],2)
    
        return jsonify(data)
             


if __name__ == '__main__':
    print('Loading model...')
    load()
    app.run(debug=False)