HinaCortus's picture
Update app.py
f44b6d9
raw
history blame
706 Bytes
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)