language_models_project/main.py

from transformers import pipeline
from transformers import AutoTokenizer, AutoModelForSequenceClassification

def classify(model_string: str, *args, **kwargs):
    tokenizer = AutoTokenizer.from_pretrained(model_string)
    model = AutoModelForSequenceClassification.from_pretrained(model_string)
    sentiment_pipeline = pipeline(
        "sentiment-analysis",
        model=model,
        tokenizer=tokenizer
    )
    return sentiment_pipeline(*args, **kwargs)