ads-video-generator / components /caption_chain.py
jonathanjordan21's picture
Create caption_chain.py
10ffaa0
raw
history blame
320 Bytes
from langchain import PromptTemplate, LLMChain
def chain(llm):
template = """Make 5 different advertisement captions about this product.
{product}
"""
prompt = PromptTemplate(template=template, input_variables=["product"])
llm_chain = LLMChain(prompt=prompt, llm=llm)
return llm_chain