Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import torch
 from transformers import AutoProcessor, UdopForConditionalGeneration
+from PIL import Image
 # from datasets import load_dataset
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -19,7 +20,8 @@ selected_file = st.selectbox("Select a document (/an image):", uploaded_files, f
 if selected_file is not None and selected_file != "None":
     file_extension = selected_file.name.split(".")[-1]
     if file_extension in ["jpg", "jpeg", "png"]:
-        image = st.image(selected_file, caption="Selected Image")
+        image = Image.open(selected_file).convert("RGB")
+        st.image(selected_file, caption="Selected Image")
     else:
         st.write("Selected file: ", selected_file.name)