Update app.py

app.py

@@ -5,31 +5,15 @@ from transformers import pipeline
 def generate_story(title, model_name):
     # Use text-generation pipeline from Hugging Face
     generator = pipeline('text-generation', model=model_name)
-    
     # Generate the story based on the input title
-    story = generator(
-        title,
-        max_length=500,  # Increase the length of the generated story to 500 tokens
-        no_repeat_ngram_size=3,  # Avoid repeating sequences of 3 words
-        temperature=0.7,  # Decrease temperature for more coherent, less random text
-        top_p=0.9,  # Consider top 90% probable words for diversity
-        top_k=40,  # Limit the next token choices to the top 40
-        repetition_penalty=1.1,  # Penalize repetition slightly to improve fluency
-        do_sample=True  # Enable sampling for creativity
-    )[0]['generated_text']
-    
-    # Clean the generated story to ensure it ends with a complete sentence
-    cleaned_story = clean_output(story)
-    
-    # Return the cleaned story
-    return cleaned_story
-
-# Function to clean the output by truncating at the last full sentence
-def clean_output(text):
-    if '.' in text:
-        return text[:text.rfind('.')+1]  # Truncate at the last full sentence
-    return text  # Return the text as is if no period is found
-
+    story = generator(title,
+                      max_length=230,  # Set the maximum length for the generated text (story) to 230 tokens
+                      no_repeat_ngram_size=3,  # Avoid repeating any sequence of 3 words (to prevent repetitive text)
+                      temperature=0.8,  # Introduce some randomness; higher values make the output more random, lower makes it more deterministic
+                      top_p=0.95  # Use nucleus sampling (top-p sampling) to focus on the top 95% of probable words, making the text more coherent
+                      )
+    # Return the generated text
+    return story[0]['generated_text']
 
 # Create the Gradio interface using gr.Interface
 demo = gr.Interface(