language_models_project/main.py

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)