Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from datasets import load_dataset
-from transformers import pipeline
+from langchain.chains import LLMChain
+from langchain.llms import HuggingFaceHub
+from langchain.prompts import PromptTemplate
 from langchain.prompts.few_shot import FewShotChatMessagePromptTemplate
 from langchain.prompts.example_selector import LengthBasedExampleSelector
 
@@ -18,8 +20,8 @@ def load_examples():
 
 examples = load_examples()
 
-# Set up the summarization model
-summarizer = pipeline("summarization", model="google/pegasus-xsum")
+# Set up the HuggingFaceHub model (T5)
+llm = HuggingFaceHub(repo_id="google/pegasus-xsum", model_kwargs={"temperature": 0.7})
 
 # Few-shot prompt template
 example_prompt = FewShotChatMessagePromptTemplate.from_examples(
@@ -31,21 +33,24 @@ example_prompt = FewShotChatMessagePromptTemplate.from_examples(
 )
 
 # Streamlit UI
-st.title("💬 Dialogue Summarizer using Few-Shot Prompt + T5")
+st.title("💬 Dialogue Summarizer using Few-Shot Prompt + T5 (via Langchain)")
 input_text = st.text_area("📝 Paste your conversation:")
 
 if st.button("Generate Summary"):
     if input_text.strip():
-        # Create prompt
+        # Create prompt using FewShotChatMessagePromptTemplate
         messages = example_prompt.format_messages(input=input_text)
 
         with st.expander("📋 Generated Prompt"):
             for msg in messages:
                 st.markdown(f"**{msg.type.upper()}**:\n```\n{msg.content}\n```")
 
-        # Generate summary using model
-        full_prompt = "\n".join([m.content for m in messages if m.type == "human"])
-        summary = summarizer("summarize: " + input_text, max_length=80, min_length=15, do_sample=False)[0]['summary_text']
+        # Create the prompt chain
+        prompt_template = PromptTemplate(input_variables=["input"], template="{input}")
+        chain = LLMChain(llm=llm, prompt=prompt_template)
+
+        # Get the summary from the model
+        summary = chain.run(input_text)
         st.success("✅ Summary:")
         st.write(summary)
     else: