Update app.py

app.py

@@ -16,6 +16,18 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained("kriton/greek-text-summarization")
 model = AutoModelForSeq2SeqLM.from_pretrained("kriton/greek-text-summarization")
 
+
+
+
+#remove incomplete sentences from the output
+def remove_incomplete_sentence(text):
+    match = re.search(r'(.+?[.!?])(?:\s|$)', text[::-1])
+    
+    if match:
+        return match.group(1)[::-1].strip()
+    else:
+        return text.strip()
+
 # Define the summary generation function
 def genarate_summary(article):
     inputs = tokenizer(
@@ -37,7 +49,7 @@ def genarate_summary(article):
         no_repeat_ngram_size=3
     )
 
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return remove_incomplete_sentence(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
 # Set up Gradio Interface
 iface = gr.Interface(