Update app.py

app.py

@@ -6,18 +6,20 @@ text_generator = pipeline("text-generation", model="gpt2")
 
 # Define the function for story generation
 def generate_story(prompt, word_count):
+    # Calculate the maximum length based on word count
+    max_length = word_count + len(prompt.split())
     # Generate a story based on the user's prompt and word count
-    generated_text = text_generator(prompt, max_length=word_count, num_return_sequences=1)[0]['generated_text']
+    generated_text = text_generator(prompt, max_length=max_length, num_return_sequences=1)[0]['generated_text']
     return generated_text
 
 # Define example inputs for the Gradio interface
 example_inputs = [
-    ["Once upon a time, in a magical forest, there was a curious rabbit named Oliver."],
-    ["Amidst the hustle and bustle of a busy city, there lived a lonely street musician."],
-    ["On a distant planet, explorers discovered an ancient alien artifact buried in the sand."],
-    ["Hidden in the attic of an old house, a forgotten journal revealed a family secret."],
-    ["In a futuristic world, a brilliant scientist invented a time-traveling device."],
-    ["Deep in the ocean, an underwater explorer encountered a mysterious and ancient creature."]
+    "Once upon a time, in a magical forest, there was a curious rabbit named Oliver.",
+    "Amidst the hustle and bustle of a busy city, there lived a lonely street musician.",
+    "On a distant planet, explorers discovered an ancient alien artifact buried in the sand.",
+    "Hidden in the attic of an old house, a forgotten journal revealed a family secret.",
+    "In a futuristic world, a brilliant scientist invented a time-traveling device.",
+    "Deep in the ocean, an underwater explorer encountered a mysterious and ancient creature."
 ]
 
 # Create a Gradio interface with examples and a word count slider
@@ -30,8 +32,8 @@ iface = gr.Interface(
     outputs="text",
     title="Story Generator with Word Count",
     description="Enter a prompt and select the word count to generate a story.",
-    examples=example_inputs
+    examples=example_inputs,
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()