Update app.py

app.py

@@ -236,7 +236,7 @@ def summ_inference(txt: str):
     txt = summ_preprocess(txt)
     inference_tokenizer, tokenized_data = summ_inference_tokenize(input_=[txt], n_tokens=SUMM_INPUT_N_TOKENS)
     pred = summ_model.generate(**tokenized_data, max_new_tokens=SUMM_TARGET_N_TOKENS)
-    result = "" if txt=="" else clean_summary(inference_tokenizer.decode(pred, skip_special_tokens=True))
+    result = "" if txt=="" else clean_summary(inference_tokenizer.decode(pred[0], skip_special_tokens=True))
     return result
 ############ SUMMARIZATION MODEL & VARS INITIALIZATION END ####################