Update app.py
Browse files
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():
|