diff --git a/summarize.py b/summarize.py index e064231..c6d137b 100755 --- a/summarize.py +++ b/summarize.py @@ -33,7 +33,7 @@ def get_summary(subtitles): response : ChatResponse = ol_client.chat(model='frowning/llama3-nymeria:15b-q6_k', messages=[ { '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', @@ -71,12 +71,7 @@ def main(): parser = argparse.ArgumentParser( description="Download subtitles from a video and summarize it using a local Ollama instance." ) - parser.add_argument( - 'url', - metavar='URL', - type=str, - help="The URL of the video to process." - ) + parser.add_argument('url', metavar='URL', type=str, help="The URL of the video to process.") # Parse out arguments args = parser.parse_args()