from transformers import pipeline from transformers import AutoTokenizer, AutoModelForSequenceClassification def classify(*args, **kwargs): tokenizer = AutoTokenizer.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis") model = AutoModelForSequenceClassification.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis") sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer) return sentiment_pipeline(*args, **kwargs)