Update main.py

main.py

@@ -31,6 +31,8 @@ async def donut_question_answering(
         # Open the image using PIL
         image = Image.open(BytesIO(contents))
 
+        image = image.convert("RGB")
+
         # Split the questions into a list
         question_list = questions.split(',')
 
@@ -45,6 +47,8 @@ async def donut_question_answering(
         return {"error": f"Error processing file: {str(e)}"}
 
 def process_document(image, questions):
+    image = image.convert("RGB")
+
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values