Update app.py

app.py

@@ -102,11 +102,11 @@ def build_model(model_choice, temperature=0.7):
         llm = HuggingFaceEndpoint(
             repo_id="google/flan-t5-base",
             huggingfacehub_api_token=api_key,
-            task="text2text-generation",
+            task="text-generation",
             max_new_tokens=500,
             temperature=temperature
         )
-        return llm, False  # (model, is_chat)
+        return ChatHuggingFace(llm=llm, temperature=temperature), True  # (model, is_chat)
 
     elif model_choice == "DeepSeek":
         llm = HuggingFaceEndpoint(
@@ -189,7 +189,7 @@ if st.button("Run Chatbot"):
                     relevant_docs = retriever.invoke(query)
                     context_text = "\n\n".join(doc.page_content for doc in relevant_docs)
 
-                prompt = prompt_template.format(context=context_text, question=query)
+                prompt = prompt_template.invoke(context=context_text, question=query)
 
                 with st.spinner(f"Generating response using {model_choice}..."):
                     model, is_chat = build_model(model_choice, temperature)