Update app.py

app.py

@@ -10,13 +10,13 @@ def generate_image_caption(image):
     result = img2caption(image)
     return result[0]['generated_text']
 
-def text2story(text): 
-    text_to_story_model = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2") 
-    story_text = text_to_story_model(text, max_new_tokens=150)[0]['generated_text'] 
-    words = story_text.split() 
-    if len(words) > 100: 
-        story_text = ' '.join(words[:100]) + '.' 
-    return story_text 
+def text2story(text):
+    text_to_story_model = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+    story_text = text_to_story_model(text, max_new_tokens=150)[0]['generated_text']
+    words = story_text.split()
+    if len(words) > 100:
+        story_text = ' '.join(words[:100]) + '.'
+    return story_text
 
 def text2speech(text):
     """Converts text to speech using a pre-trained model."""
@@ -26,10 +26,10 @@ def text2speech(text):
 
 def main():
     # App title
-    st.title("Stroyteller on Hugging Face")
+    st.title("Storyteller on Hugging Face")
     st.write("Welcome to the image to story audio app!")
 
-    uploaded_image = st.file_uploader("Upload an image(jpg, jpeg, png)", type=["jpg", "jpeg", "png"])
+    uploaded_image = st.file_uploader("Upload an image (jpg, jpeg, png)", type=["jpg", "jpeg", "png"])
 
     if uploaded_image is not None:
         image = Image.open(uploaded_image).convert("RGB")
@@ -50,7 +50,7 @@ def main():
         speech_output = text2speech(story)
         st.audio(speech_output["audio"], sample_rate=speech_output["sampling_rate"])
     else:
-        st.warning("⚠️ Unsupported file type")
+        st.warning("⚠️ Please upload an image file")
 
 if __name__ == "__main__":
     main()