| import gradio as gr | |
| import tensorflow as tf | |
| from fastai.vision.all import * | |
| # Load the model | |
| learn = tf.keras.models.load_model('sentimentality.h5') | |
| # Prediction function | |
| def make_prediction(user_sentence): | |
| prediction = learn.predict(user_sentence) | |
| dict = {'1': 'Negative', '2': 'Neutral', '3': 'Positive'} | |
| return dict[prediction[0]] | |
| title = "Sentiment Analysis MyAnimeList Reviews with fastai" | |
| description = "<p style='text-align: center'>Identifier si un commentaire dans MyAnimeList est positif, neutre ou négatif.<br/> Permet de connaître rapidement le sentiment globale que dégage un avis sur le site.</p>" | |
| app = gr.Interface(fn=make_prediction, title=title, description=description, inputs=gr.TextArea(), outputs='text') | |
| app.launch() |