Spaces:

CineAI
/

Chelsea

Sleeping

CineAI commited on Mar 29

Commit

afec314

•

1 Parent(s): 8099938

Update llm/llm.py

Files changed (1) hide show

llm/llm.py CHANGED Viewed

@@ -15,4 +15,4 @@ class LLM_chain:
         prompt = PromptTemplate(template=template, input_variables=["entity"])
         llm_chain = LLMChain(prompt=prompt, llm=self.llm, verbose=True)
         output = llm_chain.invoke(entity)
-        return output

         prompt = PromptTemplate(template=template, input_variables=["entity"])
         llm_chain = LLMChain(prompt=prompt, llm=self.llm, verbose=True)
         output = llm_chain.invoke(entity)
+        return output["text"]