Update app.py

app.py

@@ -232,11 +232,11 @@ def clean_summary(summary: str):
         summary = ""
     return summary
 
-def summ_inference(txts: str):
-    txts = [*map(summ_preprocess, txts)]
-    inference_tokenizer, tokenized_data = summ_inference_tokenize(input_=txts, n_tokens=SUMM_INPUT_N_TOKENS)
+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 t=="" else clean_summary(inference_tokenizer.decode(p, skip_special_tokens=True)) for t, p in zip(txts, pred)]
+    result = "" if txt=="" else clean_summary(inference_tokenizer.decode(pred, skip_special_tokens=True))
     return result
 ############ SUMMARIZATION MODEL & VARS INITIALIZATION END ####################