Update app.py

app.py

@@ -1,47 +1,33 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the sentiment analysis and named entity recognition pipelines
-sentiment_analysis = pipeline("sentiment-analysis")
-ner = pipeline("ner")
-
-# Create a Gradio interface
-def generate_story(user_input):
-    # Analyze the sentiment of the user's input
-    sentiment_result = sentiment_analysis(user_input)[0]
-    
-    # Identify named entities in the user's input
-    named_entities = ner(user_input)
-
-    # Generate the story based on user input, sentiment, and named entities
-    story = f"You seem to be feeling {sentiment_result['label'].lower()}.\n"
-
-    if named_entities:
-        story += "I noticed the following entities in your input:\n"
-        for entity in named_entities:
-            story += f"- {entity['word']} ({entity['entity']})\n"
-
-    story += "Once upon a time..."
-    
-    return story
-
-# Define example inputs for the interface
-example_inputs = ["I love ice cream",
-    "Tell me about Apple Inc.",
-    "Explain the concept of quantum physics",
-    "What is the capital of France?",
-    "Who is the CEO of Microsoft?",
-    "What is the population of New York City?"]
-
-# Create the Gradio interface with examples
+# Load the text generation model
+text_generator = pipeline("text-generation", model="gpt2")
+
+# Define the function for story generation
+def generate_story(prompt):
+    # Generate a story based on the user's prompt
+    generated_text = text_generator(prompt, max_length=200, num_return_sequences=1)[0]['generated_text']
+    return generated_text
+
+# Define example inputs for the Gradio interface
+example_inputs = [
+    "Once upon a time, there was a young boy who lived in a small village.",
+    "The sun was setting over the horizon, casting a warm glow over the city.",
+    "In a galaxy far, far away, a group of rebels were planning their next move.",
+    "The door creaked open, revealing a dark and mysterious room.",
+    "The clock struck midnight, and the old house came to life.",
+    "The wind howled through the trees, as the storm raged on."
+]
+
+# Create a Gradio interface with examples
 iface = gr.Interface(
     fn=generate_story,
-    inputs=gr.components.Textbox(label="Enter your input"),
-    outputs=gr.components.Textbox(label="Generated Story"),
-    live=True,
+    inputs="text",
+    outputs="text",
+    title="Story Generator",
+    description="Enter a prompt to generate a story.",
     examples=example_inputs,
-    title="Interactive Story Generator",
-    description="Enter your input, and we will generate an interactive story for you."
 )
 
 # Launch the interface