Update app.py

app.py

@@ -16,11 +16,15 @@ def img2text(img):
 
 # text2story
 def text2story(text):
-    generator = pipeline("text-generation",
-                     model="distilbert/distilgpt2")
-    story_text = generator(text,
-                            min_length=100,
-                            num_return_sequences=1)[0]["generated_text"]
+    generator = pipeline("text-generation", model="distilbert/distilgpt2")
+    story_text = generator(
+        text,
+        min_length=150,  # min_length, # of tokens at least larger than100
+        max_length=300,  
+        num_return_sequences=1,
+        no_repeat_ngram_size=2,  # prevent repetition
+        early_stopping=False     # prohibit early stopping
+    )[0]["generated_text"]
     return story_text