Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+
+# ✅ UPDATED imports
 from langchain_community.llms import HuggingFacePipeline
-from langchain.prompts import PromptTemplate
+from langchain_core.prompts import PromptTemplate
 from langchain.chains import LLMChain
 
 # -------------------------
-# Model config (CPU safe)
+# Model (CPU safe)
 # -------------------------
 MODEL_NAME = "google/flan-t5-base"
 
@@ -13,7 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 
 hf_pipeline = pipeline(
-    task="text2text-generation",
+    "text2text-generation",
     model=model,
     tokenizer=tokenizer,
     max_length=256,
@@ -29,7 +31,7 @@ prompt = PromptTemplate(
     input_variables=["question"],
     template="""
 You are an intelligent AI assistant.
-Answer clearly and concisely.
+Answer the question clearly.
 
 Question: {question}
 Answer:
@@ -51,10 +53,11 @@ def chat_fn(user_input):
 # -------------------------
 demo = gr.Interface(
     fn=chat_fn,
-    inputs=gr.Textbox(lines=2, placeholder="Ask your question here..."),
+    inputs=gr.Textbox(lines=2, placeholder="Ask your question..."),
     outputs="text",
     title="LangChain + Hugging Face Chatbot",
-    description="Deployed on Hugging Face Spaces (CPU)"
+    description="Running on Hugging Face Spaces (LangChain v0.1+ compatible)"
 )
 
 demo.launch()
+