Spaces:
Sleeping
Sleeping
changed function to run on cpu
Browse files
TranscriptApi/common/utils.py
CHANGED
@@ -129,7 +129,7 @@ def summarize_youtube_video(video_url, outputs_dir):
|
|
129 |
[INFO] num_steps : {num_steps}
|
130 |
""")
|
131 |
|
132 |
-
summarizer = pipeline('summarization', model = checkpoint, tokenizer = checkpoint, max_length = 200, truncation = True
|
133 |
|
134 |
summaries = []
|
135 |
|
|
|
129 |
[INFO] num_steps : {num_steps}
|
130 |
""")
|
131 |
|
132 |
+
summarizer = pipeline('summarization', model = checkpoint, tokenizer = checkpoint, max_length = 200, truncation = True)
|
133 |
|
134 |
summaries = []
|
135 |
|