Update app.py

app.py

@@ -19,7 +19,7 @@ def load_llm(max_tokens, prompt_template, temperature):
         temperature=temperature,
     )
 
-    llm_chain = LLMChain(llm=llm, prompt=PromptTemplate.from_template(prompt_template))
+    llm_chain = LLMChain(llm=llm, prompt=prompt_template)
     return llm_chain
 
 def get_src_original_url(query):
@@ -86,9 +86,8 @@ def main():
                 st.write("Image of the article is: " + image_input)
 
                 prompt_template = example_prompt_template.format(user_input=user_input)
-                prompt_dict = {'text': prompt_template}  # Convert the prompt to a dictionary
-                llm_call = load_llm(max_tokens=max_tokens, prompt_template=prompt_dict, temperature=temperature)
-                result = llm_call({'text': user_input})  # Pass user input as a dictionary
+                llm_call = load_llm(max_tokens=max_tokens, prompt_template=prompt_template, temperature=temperature)
+                result = llm_call(user_input)
                 if len(result) > 0:
                     st.info("Your article has been generated successfully!")
                     st.write(result['text'])