Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from transformers import pipeline
 from io import BytesIO
+from PIL import Image
 import torch
 
 # Load pipelines
@@ -13,11 +14,11 @@ st.title("Image-to-Text and Text-to-Speech App")
 uploaded_image = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
 
 if uploaded_image:
-    st.image(uploaded_image, caption="Uploaded Image", use_column_width=True)
+    image = Image.open(uploaded_image)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
     
     # Convert image to text
-    image_bytes = uploaded_image.read()
-    text_output = image_to_text(image_bytes)[0]['generated_text']
+    text_output = image_to_text(image)[0]['generated_text']
     st.write("### Extracted Text:")
     st.write(text_output)