app.py

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]]
  return prediction

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()