app.py

from shiny import App, Inputs

from transformers import AutoTokenizer, AutoModelForSequenceClassification
from transformers import TextClassificationPipeline

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


app = App(
    print('Processing')
    Inputs = "Le Bayern Munich prend la grenadine."
    classification(Inputs)
)