Update app.py

app.py

@@ -12,8 +12,8 @@ def summarize(text):
     summarized_ids = model.generate(inputs['input_ids'], attention_mask=inputs['attention_mask'], 
                                     max_length=150, num_beams=4, early_stopping=True)
 
-    return " ".join([tokenizer.decode(token_ids, skip_special_tokens=True)
-                    for token_ids in summarized_ids])
+    return tokenizer.decode(summarized_ids[0], skip_special_tokens=True)
+