Spaces:
Runtime error
Runtime error
from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
from transformers import TextClassificationPipeline | |
from shiny import App, Inputs | |
def classification(input: Inputs): | |
model_name = 'lincoln/flaubert-mlsum-topic-classification' | |
loaded_tokenizer = AutoTokenizer.from_pretrained(model_name) | |
loaded_model = AutoModelForSequenceClassification.from_pretrained(model_name) | |
nlp = TextClassificationPipeline(model=loaded_model, tokenizer=loaded_tokenizer) | |
result = nlp(Inputs, truncation=True) | |
print(result) | |
return result | |
if "main" in "__main__": | |
print('Processing') | |
Inputs = "Le Bayern Munich prend la grenadine." | |
classification(Inputs) | |