Update app.py

app.py

@@ -33,13 +33,8 @@ def predict(image, prompt=None):
     result = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
     return result
 
-Hàm mock xử lý ảnh — chỉ để test UI
-def mock_predict(image, prompt=None):
-    return f"Fake OCR result for image. Prompt: {prompt or 'N/A'}"
-
 demo = gr.Interface(
-    # fn=predict,
-    fn=mock_predict,
+    fn=predict,
     inputs=[
         gr.Image(type="pil", label="Tải ảnh tài liệu lên"),
         gr.Textbox(label="Gợi ý (tuỳ chọn)", placeholder="VD: Trích số hóa đơn")
@@ -48,5 +43,5 @@ demo = gr.Interface(
     title="Camel-Doc OCR - Trích xuất văn bản từ ảnh"
 )
 
-# if __name__ == "__main__":
-#     demo.launch()
+if __name__ == "__main__":
+    demo.launch()