Update app.py

app.py

@@ -15,14 +15,20 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-
-    child_prompt = f"Generate a children's story based on: {text}"
-    
-    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    
-    story_text = pipe(child_prompt,
-                     max_length=150,
-                     num_return_sequences=1)[0]['generated_text']
+    pipe = pipeline(
+        "text-generation", 
+        model="pranavpsv/genre-story-generator-v2",
+        max_new_tokens=300,            
+        min_new_tokens=100,
+        do_sample=True,
+        temperature=0.65,             
+        top_p=0.85,
+        top_k=30,                     
+        repetition_penalty=1.3,
+        no_repeat_ngram_size=3,       
+        early_stopping=False          
+    )
+    story_text = pipe(text, return_full_text=False)[0]['generated_text']
     return story_text