Spaces:
Runtime error
Runtime error
import gradio as gr | |
import transformers | |
from transformers import AutoModelForSequenceClassification, AutoTokenizer | |
from transformers import pipeline | |
model = AutoModelForSequenceClassification.from_pretrained('mrcoombes/distilbert-wikipedia-pokemon') | |
tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased') | |
classifier = pipeline('text-classification', model = model, tokenizer=tokenizer, return_all_scores=True) | |
clf = lambda x: sorted(classifier(x)[0], key=lambda y: y['score'], reverse=True) | |
demo = gr.Interface(fn=clf, inputs="text", outputs="text") | |
demo.launch() |