Update app.py

app.py

@@ -1,12 +1,14 @@
 # import part - only using the two requested imports
 import streamlit as st
 from transformers import pipeline
+from PIL import Image
+import io
 
 # function part
 # img2text
-def img2text(image_path):
+def img2text(image):
     image_to_text = pipeline("image-to-text", model="sooh-j/blip-image-captioning-base")
-    text = image_to_text(image_path)[0]["generated_text"]
+    text = image_to_text(image)[0]["generated_text"]
     return text
 
 # text2story - IMPROVED to end naturally
@@ -91,12 +93,12 @@ if uploaded_file is not None:
     # Display the uploaded image
     st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
     
-    # Create a temporary file in memory from the uploaded file
-    image_bytes = uploaded_file.getvalue()
+    # Convert the file to a PIL Image
+    image = Image.open(uploaded_file)
     
     # Stage 1: Image to Text
     st.text('Processing img2text...')
-    caption = img2text(image_bytes)  # Pass bytes directly to pipeline
+    caption = img2text(image)  # Pass PIL image to pipeline
     st.write(caption)
     
     # Stage 2: Text to Story