update

utils/model.py

@@ -57,4 +57,4 @@ class Model(torch.nn.Module):
             eos_token_id=self.tokenizer.eos_token_id,
         )
         
-        return sequences[-1]['generated_text']
+        return sequences[-1]['summary_text']