Update app.py

app.py

@@ -22,11 +22,12 @@ def process_image(image):
     else:
         image = image.convert("RGB")
 
-    # Encode và predict
+    # OCR
     pixel_values = processor(image, return_tensors="pt").pixel_values
     generated_ids = model.generate(pixel_values)
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    return generated_text
+
+    return image, generated_text
 
 # Giao diện Gradio
 title = "Captcha OCR Demo"
@@ -35,11 +36,12 @@ description = "Nhận diện captcha từ mã số thuế (MST) – Model TrOCR"
 interface = gr.Interface(
     fn=process_image,
     inputs="image",
-    outputs="text",
+    outputs=["image", "text"],
     examples=[f"examples/captcha-{i}.png" for i in range(10)],
-    title=title,
-    description=description,
+    title="Captcha OCR Demo",
+    description="Xem ảnh đã xử lý (nền trắng) và kết quả OCR",
 )
 
+
 if __name__ == "__main__":
     interface.launch()