Update app.py

app.py

@@ -8,15 +8,13 @@ processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-printed")
 
 
 def process_image(image):
-    # Chuyển ảnh sang RGB và thêm nền trắng nếu ảnh có alpha
-    image = image.convert("RGBA")
-    background = Image.new("RGBA", image.size, (255, 255, 255, 255))  # trắng hoàn toàn
-    image = Image.alpha_composite(background, image).convert("RGB")   # merge và chuyển sang RGB
+    # Chuyển nền trong suốt thành trắng (nếu có alpha channel)
+    if image.mode in ("RGBA", "LA"):
+        background = Image.new("RGB", image.size, (255, 255, 255))
+        image = Image.alpha_composite(background, image.convert("RGBA"))
 
-    # Chuẩn bị ảnh cho model
+    # Xử lý ảnh
     pixel_values = processor(image, return_tensors="pt").pixel_values
-
-    # Sinh kết quả
     generated_ids = model.generate(pixel_values)
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
     return generated_text
@@ -32,6 +30,4 @@ interface = gr.Interface(
     outputs="text",
     title=title,
     description=description
-)
-
-interface.launch()
+).launch()