models.py

from transformers import pipeline

def load_model():
    return pipeline("text-generation", model="deepseek-ai/DeepSeek-R1")