Update llama model URL in summarizer.py and translator.py

This commit is contained in:
Mert Cobanov 2024-04-26 05:04:54 +03:00
parent 97918fed73
commit 5a2b0e1e23
2 changed files with 2 additions and 2 deletions

View File

@ -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

View File

@ -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