Fix prompt template

app.py

@@ -33,7 +33,9 @@ executor = ThreadPoolExecutor(max_workers=2)
 
 
 def generate_base_response(input_text):
-    return base_pipe(input_text)[0]["generated_text"]
+    formatted_input = prompt_template.format(input_text=input_text)
+    result = base_pipe(formatted_input)[0]["generated_text"]
+    return result.split("<|assistant|>")[1].strip()
 
 
 def generate_irai_response(input_text):
@@ -58,7 +60,7 @@ def generate_response(input_text):
 st.title("Base Model vs IRAI LLM-ADE")
 user_input = st.text_area("Enter a financial question:", "")
 
-if st.button("Generate"):
+if st.button("Generate") or user_input:
     if user_input:
         base_response, irai_response = generate_response(user_input)
         col1, col2 = st.columns(2)