Update app.py

app.py

@@ -8,13 +8,13 @@ from transformers import pipeline
 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"]
+    text = image_to_text_model(img_path)[0]["generated_text"]
     return text
 
 # text2story
 def text2story(text):
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    story_text = pipe(text)[0]['generated_text']
+    story_text = pipe(text, max_length=100)[0]['generated_text']
     return story_text
 
 # text2audio
@@ -29,7 +29,7 @@ def text2audio(story_text):
 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...")
+uploaded_file = st.file_uploader("Select an Image...", type=["jpg", "jpeg", "png"])
 
 
 if uploaded_file is not None: