import pandas as pd
from fastai.text.all import *
from datasets import load_dataset

learn = load_learner('finetunedClass.pth')

def predict(frase):
   pred = learn.predict(frase)
   return pred
   
gr.Interface(fn=predict, inputs="text", outputs="text", examples=['i had been talking to coach claudia barcomb and coach ali boe for a long time and they both made me feel very welcomed at union']).launch(share=False)