| import gradio as gr | |
| from huggingface_hub import from_pretrained_fastai | |
| from blurr.text.modeling.all import * | |
| def predict(text): | |
| repo_id = "msivanes/blurr_IMDB_distilbert_cls" | |
| learn = from_pretrained_fastai(repo_id) | |
| probs = learn.predict(text) | |
| return f"Sentiment Prediction Confidence for the sentence '{text}' as positive is {100 * probs[0]['probs'][1]:.2f}%" | |
| examples = ['This hack day is amazing', 'I hate the weather today.', 'Loved how Ryan started talking in gibberish language'] | |
| demo = gr.Interface(fn=predict, inputs=gr.Textbox(lines=3, label="Input Text"), outputs="text", title='DistiBERT Sentiment Classification', examples=examples) | |
| demo.launch() |