diff --git a/summarizer.py b/summarizer.py index bd59acd..9dbcf44 100644 --- a/summarizer.py +++ b/summarizer.py @@ -45,7 +45,7 @@ def setup_summarization_chain(): input_variables=["text"], ) - llm = ChatOllama(model="llama3") + llm = ChatOllama(model="llama3", base_url="http://0.0.0.0:11434") llm_chain = LLMChain(llm=llm, prompt=prompt_template) return llm_chain diff --git a/translator.py b/translator.py index 3601718..816fbef 100644 --- a/translator.py +++ b/translator.py @@ -14,6 +14,6 @@ def setup_translator_chain(): input_variables=["text"], ) - llm = ChatOllama(model="llama3") + llm = ChatOllama(model="llama3", base_url="http://0.0.0.0:11434") llm_chain = LLMChain(llm=llm, prompt=prompt_template) return llm_chain