Create app.py

app.py

@@ -0,0 +1,48 @@
+# import part
+import streamlit as st
+from transformers import pipeline
+from PIL import Image
+
+# function part
+# image to text
+def img2text(img):
+    image_to_text_model = pipeline("image-to-text",
+                                   model="Salesforce/blip-image-captioning-base")
+    text = image_to_text_model(img)[0]["generated_text"]
+    return text
+    st.write(text)
+
+# text to story
+def text2story(text):
+    text_generation_model = pipeline("text-generation",
+                                   model="meta-llama/Meta-Llama-3-8B")
+    story_text = "Once upon a time in a land far, far away"
+    generated_story = story_text(story_text,
+                            max_length=100,
+                            num_return_sequences=1)
+
+    return generated_story
+    st.write(generated_story)
+
+# text to audio
+def text2audio(story_text)
+    text_to_speech_model = pipeline("text-to-speech", model="facebook/mms-tts-eng")
+    text_to_speak = """
+    Hi everyone, I'm going to start with a story now！
+    """
+    speech_output = text_to_speech_model(text_to_speak)
+    print("Text-to-Speech Output:", speech_output)
+
+from IPython.display import Audio
+    print(text_to_speak)
+    st.audio(speech_output['audio'], 
+             sample_rate=speech_output['sampling_rate'])
+
+# main part
+st.set_page_config(page_title="Your Image to Audio Story",
+                    page_icon="*")
+st.header("Turn Your Image to Audio Story")
+uploaded_file = st.file_uploader("Select an Image...", type=["jpg", "png", "jpeg"])
+if uploaded_file is not None:
+    image = Image.open(uploaded_file).convert("RGB")
+    st.image(image, caption="Uploaded Image", use_column_width=True)