Update app.py

Added PIL processing

app.py

@@ -1,9 +1,14 @@
 import gradio as gr
 from transformers import VisionEncoderDecoderModel, TrOCRProcessor
 import torch
+from PIL import Image
 
 def recognize_captcha(input, mdl):
-    input = input.convert('RGB')
+    # Load image
+    image = Image.open(input).convert("RGB")
+    # Create white background
+    background = Image.new("RGBA", image.size, (255, 255, 255))
+    combined = Image.alpha_composite(background, image).convert("RGB")
 
     # Load model and processor
     processor = TrOCRProcessor.from_pretrained(mdl)