Hugging Face's logo

Spaces:
asataura
/
jam_shield_LLM_app
Sleeping

App Files Community
asataura commited on
Commit
d2b8088
·
1 Parent(s): 88cee09

installing langchain

Browse files
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -20,7 +20,7 @@ llm = HuggingFaceHub(huggingfacehub_api_token=huggingfacehub_api_token,
20
  template = """You are an AI trained to analyze and provide insights about training graphs in the domain of deep
21
  reinforcement learning. Given the following data about a graph: {data}, provide detailed insights. """
22
 
23
- prompt = PromptTemplate(template=template, input_variables=["question"])
24
  llm_chain = LLMChain(prompt=prompt, verbose=True, llm=llm)
25
 
26
 
 
20
  template = """You are an AI trained to analyze and provide insights about training graphs in the domain of deep
21
  reinforcement learning. Given the following data about a graph: {data}, provide detailed insights. """
22
 
23
+ prompt = PromptTemplate(template=template, input_variables=["data"])
24
  llm_chain = LLMChain(prompt=prompt, verbose=True, llm=llm)
25
 
26