hassaanik's picture
Upload 8 files
24bf069 verified
from flask import Flask, request, render_template, jsonify
from predict import predict_language
import joblib
import tensorflow as tf
import h5py
model = tf.keras.models.load_model('models\\full_language_identifcation_modelf.h5')
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
CountVectorizer = joblib.load('models\\cv.joblib')
LabelEncoder = joblib.load('models\\le.joblib')
app = Flask(__name__)
@app.route('/', methods=['GET', 'POST'])
def predict():
if request.method == 'POST':
text = request.form['text']
prediction = predict_language(text, model, CountVectorizer, LabelEncoder) # Call your prediction function
return render_template('result.html', prediction=prediction, text=text)
return render_template('index.html')
if __name__ == '__main__':
app.run(debug=True)