s
File size: 246 Bytes
e8a4c92
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
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"]