diff --git a/summarizer.py b/summarizer.py index 8736d09..db12eff 100644 --- a/summarizer.py +++ b/summarizer.py @@ -1,7 +1,6 @@ import argparse from langchain.chains.llm import LLMChain -from langchain.chains.summarize import load_summarize_chain from langchain.prompts import PromptTemplate from langchain_community.chat_models import ChatOllama from langchain_community.document_loaders import WebBaseLoader diff --git a/summerizer_concise.py b/summerizer_concise.py index 30e2d5d..64d0133 100644 --- a/summerizer_concise.py +++ b/summerizer_concise.py @@ -1,7 +1,6 @@ import argparse from langchain.chains.llm import LLMChain -from langchain.chains.summarize import load_summarize_chain from langchain.prompts import PromptTemplate from langchain_community.chat_models import ChatOllama from langchain_community.document_loaders import WebBaseLoader diff --git a/webui.py b/webui.py index ea1b2f6..e7cc77b 100644 --- a/webui.py +++ b/webui.py @@ -1,5 +1,6 @@ import gradio as gr -from summarizer import setup_summarization_chain, load_document + +from summarizer import load_document, setup_summarization_chain def summarize(url):