Update app.py

app.py

@@ -1,52 +1,12 @@
 import streamlit as st
-from transformers import pipeline
-
-# Load Hugging Face pipelines
-text_generator = pipeline("text-generation")
-text_to_speech = pipeline("text-to-speech")
-text_to_image = pipeline("text2image")
-
-# Streamlit app
-st.title("Children's Storytelling App")
 
-# Input prompt for story
-story_prompt = st.text_area("Write the beginning of your story:")
-
-# Generate story button
-if st.button("Generate Story"):
-    if story_prompt:
-        # Generate text
-        generated_story = text_generator(story_prompt, max_length=100, num_return_sequences=1)
-        st.write("Here's your story:")
-        st.write(generated_story[0]["generated_text"])
-    else:
-        st.warning("Please enter a story prompt.")
+from transformers import pipeline
 
-# Text-to-speech button
-if st.button("Text to Speech"):
-    if story_prompt:
-        # Convert text to speech
-        st.audio(text_to_speech(story_prompt)[0]["audio"], format="audio/wav")
-    else:
-        st.warning("Please enter a story prompt.")
+text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=0)  # You can choose a different model if needed
 
-# Text-to-text button
-if st.button("Text to Text"):
-    if story_prompt:
-        # Convert text to a different text
-        transformed_text = text_to_image(story_prompt)[0]["generated_text"]
-        st.write("Transformed text:")
-        st.write(transformed_text)
-    else:
-        st.warning("Please enter a story prompt.")
+# Generate text
+prompt = "Once upon a time"
+generated_text = text_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
 
-# Text-to-image
-st.sidebar.title("Text to Image")
-image_prompt = st.sidebar.text_area("Enter text for image:")
-if st.sidebar.button("Generate Image"):
-    if image_prompt:
-        # Convert text to image
-        image = text_to_image(image_prompt)[0]["image"]
-        st.sidebar.image(image, caption="Generated Image", use_column_width=True)
-    else:
-        st.sidebar.warning("Please enter text for the image.")
+# Print the generated text
+print(generated_text)