Arman511's picture
Update app.py
eaafae6
raw
history blame
342 Bytes
from fastai.text.all import *
import gradio as gr
learn.load_learner('export.pkl')
categories = ('Negative', 'Positive')
label = gr.outputs.Label()
def classify(text):
pred,pred_idx,probs = learn.predict(text)
return {labels[i]: float(probs[i]) for i in range(len(labels))}
intf = gr.Interface(classify,"text",label)
intf.launch()