From 5a2b0e1e230932088ac81daaf5dffb19c6f62bd5 Mon Sep 17 00:00:00 2001 From: Mert Cobanov Date: Fri, 26 Apr 2024 05:04:54 +0300 Subject: [PATCH] Update llama model URL in summarizer.py and translator.py --- summarizer.py | 2 +- translator.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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