tcsenpai/easy-web-summarizer
1
0
Fork 0
mirror of https://github.com/tcsenpai/easy-web-summarizer.git synced 2025-06-06 18:45:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update base_url in summarizer.py, translator.py, and yt_summarizer.py

Browse Source
This commit is contained in:
Mert Cobanov 2024-05-01 19:38:48 +03:00
parent 935a605f6b
commit 10923d535e
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
summarizer.py
View File

@ -45,7 +45,7 @@ def setup_summarization_chain():
input_variables=["text"],
)
llm = ChatOllama(model="llama3", base_url="http://127.0.0.1:11434")
llm = ChatOllama(model="llama3:instruct", base_url="http://127.0.0.1:11434")
llm_chain = LLMChain(llm=llm, prompt=prompt_template)
return llm_chain

2
translator.py
View File

@ -14,6 +14,6 @@ def setup_translator_chain():
input_variables=["text"],
)
llm = ChatOllama(model="llama3", base_url="http://127.0.0.1:11434")
llm = ChatOllama(model="llama3:instruct", base_url="http://127.0.0.1:11434")
llm_chain = LLMChain(llm=llm, prompt=prompt_template)
return llm_chain

2
yt_summarizer.py
View File

@ -48,7 +48,7 @@ def yt_summarization_chain():
DETAILED SUMMARY:""",
input_variables=["text"],
)
llm = ChatOllama(model="llama3", base_url="http://127.0.0.1:11434")
llm = ChatOllama(model="llama3:instruct", base_url="http://127.0.0.1:11434")
summarize_chain = load_summarize_chain(
llm=llm, prompt=prompt_template, verbose=True
)