| import gradio as gr | |
| import pickle | |
| import pandas as pd | |
| file = open("https://github.com/lorenzoscottb/PLANE/blob/main/supervised_learning/OoD/OOV_Train_2.pkl","rb") | |
| data = pickle.load(file) | |
| description = "Did you know that, logically speaking, a small cat is not a small animal, and that a fake smile is not a smile. \n Learn more testing our BERT model is tuned to perform phrase-level adjective-noun entailment, via the [PLANE](https://aclanthology.org/2022.coling-1.359/) dataset. \n The current model achieves an accuracy of 90% on out-of-distribution evaluation" | |
| title = "BERT on a PLANE" | |
| examples = [["A red car is a vehicle"], ["A fake smile is a smile"], ["A small cat is a small animal"]] | |
| interface = gr.Interface.load("huggingface/lorenzoscottb/bert-base-cased-PLANE-ood-2", | |
| description=description, | |
| examples=examples, | |
| title=title, | |
| ) | |
| interface.launch() |