added function to convert image to RGB

app.py

@@ -16,12 +16,13 @@ def TrOCR_predict(pixel_values, processor, model):
     return predicted_text
 
 
-df = pd.DataFrame()
 uploaded_file = st.file_uploader("Choose a file")
 if uploaded_file is not None:
     content = uploaded_file.read()
     st.image(uploaded_file)
     image = Image.open(uploaded_file)
+    if image.mode != "RGB": # Convert the image to RGB
+        image = image.convert("RGB")
     pixel_values = processor(images=image, return_tensors="pt").pixel_values
     predicted_text = TrOCR_predict(pixel_values, processor, model)[0]
     texts = predicted_text