salt/summarizeyt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor formatting and prompt tweaks

Browse Source
This commit is contained in:
Salt
2025-01-08 13:52:06 -06:00
parent fca8c6e232
commit 54980ae706
1 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
summarize.py
View File

@@ -33,7 +33,7 @@ def get_summary(subtitles):
response : ChatResponse = ol_client.chat(model='frowning/llama3-nymeria:15b-q6_k', messages=[ response : ChatResponse = ol_client.chat(model='frowning/llama3-nymeria:15b-q6_k', messages=[
{ {
'role': 'system', 'role': 'system',
'content': 'Your job is to summarize YouTube videos given a (potentially auto-generated) transcript. Summarize the video, cutting out sponsor segments and advertisements. Include all core points in the video. Be as detailed as possible.' 'content': 'Your job is to summarize YouTube videos given a (potentially auto-generated) transcript. Summarize the video, cutting out sponsor segments and advertisements. Include all core points in the video. Be as detailed as possible. Your response should be at least five paragraphs.'
}, },
{ {
'role': 'user', 'role': 'user',
@@ -71,12 +71,7 @@ def main():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description="Download subtitles from a video and summarize it using a local Ollama instance." description="Download subtitles from a video and summarize it using a local Ollama instance."
) )
parser.add_argument( parser.add_argument('url', metavar='URL', type=str, help="The URL of the video to process.")
'url',
metavar='URL',
type=str,
help="The URL of the video to process."
)
# Parse out arguments # Parse out arguments
args = parser.parse_args() args = parser.parse_args()