Kokoro-API-2

Sleeping

Yaron Koresh commited on Jan 21

Commit

966bdcc

verified ·

1 Parent(s): 723fc44

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -569,7 +569,7 @@ def summarize_text(
             num_beams=4,
             early_stopping=True
         )
-        inputs[0] = [*list(outputs[0]), *list(inputs[0][512:])]
         i = i + 1
     summary = tokenizer.decode(inputs[0])
     log(f'RET summarize_text with summary as {summary}')

             num_beams=4,
             early_stopping=True
         )
+        inputs = torch.tensor([[*list(outputs[0]), *list(inputs[0][512:])]])
         i = i + 1
     summary = tokenizer.decode(inputs[0])
     log(f'RET summarize_text with summary as {summary}')