Update app.py

app.py

@@ -1,15 +1,25 @@
 from transformers import pipeline, set_seed
 import gradio as gr
+import nltk
+
+nltk.download('punkt')
 
 classifier = pipeline('text-generation', model='arputtick/GPT_Neo_1.3B_eco_feminist_2')
 set_seed(42)
 
 def generate_text(text, gen_length):
     gen_text = classifier(text, max_length=gen_length)[0]['generated_text']
-    return gen_text
+    sentences = nltk.sent_tokenize(gen_text)
+
+    if sentences[-1][-1] == ".":
+        output = sentences
+
+    else:
+        output = sentences[:-1]
+    return output
 
 Instructuction = "Browse the internet to download any unique image"
-title="Eco-Feminists Text Generation"
+title="Eco-Feminist Text Generation"
 description = "Start writing a peice of text in the input box\
                and see how well the text generation language model\
                is able to generate new text that uniquely completes your sentences."