fdfdffd

app.py

@@ -1,15 +1,14 @@
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import JSONResponse
-from magic_pdf import MinerU
+from magic_pdf.api.magic_pdf_api import process_pdf
 
 app = FastAPI()
-model = MinerU(config_path="/root/magic-pdf.json")
 
 @app.post("/extract")
 async def extract(file: UploadFile = File(...)):
     content = await file.read()
     try:
-        result = model.extract(content)
+        result = process_pdf(pdf_bytes=content, config_path="/root/magic-pdf.json")
         return {"result": result}
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})