llm_utils.py

from transformers import pipeline

# Load the Hugging Face LLM
llm = pipeline("text-generation", model="gpt2", max_length=100)

def get_llm_response(prompt: str) -> str:
    response = llm(prompt)
    return response[0]["generated_text"]