Fix/gradio demo

marker_api/demo.py

@@ -53,24 +53,23 @@ def parse_document(input_file_path, parameters, request: gr.Request):
             mime_type = "application/octet-stream"  # Default MIME type if not found
 
         with open(input_file_path, "rb") as f:
-            files = {"file": (input_file_path, f, mime_type)}
+            files = {"pdf_file": (input_file_path, f, mime_type)}
             response = requests.post(
                 post_url, files=files, headers={"accept": "application/json"}
             )
 
         document_response = response.json()
-
-        images = document_response.get("images", [])
-
+        result = document_response.get("result", {})
+        markdown_text = str(result.get("markdown", ""))
+        images = result.get("images", {})
+
         # Decode each base64-encoded image to a PIL image
-        pil_images = [
-            decode_base64_to_pil(image_dict["image"]) for image_dict in images
-        ]
-
+        pil_images = [decode_base64_to_pil(img) for img in images.values()]
+
         return (
-            str(document_response["text"]),
+            markdown_text,
             gr.Gallery(value=pil_images, visible=True),
-            str(document_response["text"]),
+            markdown_text,
             gr.JSON(value=document_response, visible=True),
         )
 

pyproject.toml

@@ -23,6 +23,7 @@ marker-pdf = "^0.2.17"
 pynvml = "^11.5.3"
 art = "^6.3"
 gradio = "^5.1.0"
+transformers = "4.45.2"
 
 
 

server.py

@@ -52,9 +52,6 @@ async def lifespan(app: FastAPI):
     allow_credentials=True,
 )
 
-app = gr.mount_gradio_app(app, demo_ui, path="")
-
-
 @app.get("/health", response_model=HealthResponse)
 def server():
     """
@@ -97,6 +94,8 @@ async def process_files(files):
     responses = await process_files(pdf_files)
     return BatchConversionResponse(results=responses)
 
+# Mount the Gradio app last as the other routes are unavailable otherwise
+gr.mount_gradio_app(app, demo_ui, path="")
 
 # Main function to run the server
 def main():