Th3BossC commited on
Commit
c1ddee8
·
1 Parent(s): 68c70b0

changed function to run on cpu

Browse files
Files changed (1) hide show
  1. TranscriptApi/common/utils.py +1 -1
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, device = 0)
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