Update model.py

src/models/model.py

@@ -458,9 +458,8 @@ class Summarization:
             top_k=top_k,
             num_return_sequences=num_return_sequences,
         )
-        preds = [
-            self.tokenizer.decode(
-                g,
+        preds = self.tokenizer.decode(
+                generated_ids[0],
                 skip_special_tokens=skip_special_tokens,
                 clean_up_tokenization_spaces=clean_up_tokenization_spaces,
             )