From 54980ae7068b8b914c89877a40bbe562e50bb487 Mon Sep 17 00:00:00 2001
From: Jacob Babor <jacob@babor.tech>
Date: Wed, 8 Jan 2025 13:52:06 -0600
Subject: [PATCH] Minor formatting and prompt tweaks
---
summarize.py | 9 ++-------
1 file changed, 2 insertions(+), 7 deletions(-)
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()