sooolee commited on
Commit
c04214b
·
1 Parent(s): 3748e4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def summarize(video_id):
51
  transcript += dict[i]['text']
52
 
53
  texts = preprocessing(transcript)
54
- inputs = tokenizer(texts, return_tensors="pt", padding=True, )
55
  inputs = inputs["input_ids"].to(device)
56
 
57
  with torch.no_grad():
 
51
  transcript += dict[i]['text']
52
 
53
  texts = preprocessing(transcript)
54
+ inputs = tokenizer(*texts, return_tensors="pt", padding=True, )
55
  inputs = inputs["input_ids"].to(device)
56
 
57
  with torch.no_grad():