Update app.py

app.py

@@ -1,12 +1,17 @@
+# Install necessary libraries
+!pip install -q gradio
+!pip install -q transformers
+!pip install -q gtts
 
 # Import libraries
 import gradio as gr
-import openai
+from transformers import pipeline
 from gtts import gTTS
 from io import BytesIO
 from IPython.display import Audio
 
-# Set up OpenAI API key
+# Create a text generation pipeline with GPT-2
+story_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
 
 # Set the context for the storyteller
 messages = [{"role": "system", "content": "You are a magical storyteller, creating wonderful tales for kids. Make them imaginative and full of joy!"}]
@@ -14,12 +19,10 @@ messages = [{"role": "system", "content": "You are a magical storyteller, creati
 # Define the Storyteller function
 def StorytellerGPT(tell_story):
     messages.append({"role": "user", "content": tell_story})
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages,
-        temperature=.1
-    )
-    story_reply = response["choices"][0]["message"]["content"]
+    
+    # Generate story using Hugging Face's GPT-2
+    story_reply = story_generator(tell_story, max_length=100, num_return_sequences=1)[0]['generated_text']
+    
     messages.append({"role": "assistant", "content": story_reply})
 
     # Convert text to speech
@@ -39,4 +42,5 @@ demo = gr.Interface(
     description="A magical storyteller app for kids! Type a sentence, and let the app create an enchanting story for you."
 )
 
-# Launch the Gradio 
+# Launch the Gradio Interface
+demo.launch()