Update model.py

src/models/model.py

@@ -450,3 +450,4 @@ class Summarization:
             )
             for g in generated_ids
         ]
+        return preds[0]