| import gradio as gr | |
| import os | |
| from transformers import BertTokenizer, BertModel | |
| import torch | |
| tokenizer = BertTokenizer.from_pretrained('bert-base-uncased') | |
| def Bemenet(bemenet): | |
| return tokenizer.tokenize("[CLS] " + bemenet + " [SEP]") | |
| interface = gr.Interface(fn=Bemenet, | |
| title="Cím..", | |
| description="Leírás..", | |
| inputs="text", | |
| outputs="text") | |
| interface.launch() |