Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 from langchain.prompts import PromptTemplate
-from langchain.chains import RunnableSequence
+from langchain.chains import LLMChain
 
 
 # Inicializa o pipeline do Hugging Face
@@ -34,15 +34,12 @@ prompt = PromptTemplate(
     input_variables=["us", "ca", "lp", "fw"],
     template=prompt_template
 )
-chain_1 = RunnableSequence(prompt | llm)
+chain_1 = LLMChain(llm=llm, prompt=prompt)
 
 if st.button("Generate Unit Tests"):
     # Gera o resultado usando os valores de entrada
     inputs = {"us": us, "ca": ca, "lp": lp, "fw": fw}
-    formatted_prompt = prompt.format(inputs)
-    
-    # Utiliza o modelo da Hugging Face para gerar o texto
-    result = chain_1.run(formatted_prompt)
+    result = chain_1.run(inputs)
     
     # Exibe o resultado
     st.write("Results:")