File size: 472 Bytes
44df236
 
a217d36
44df236
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from fastapi import FastAPI, UploadFile, File
from fastapi.responses import JSONResponse
from magic_pdf import MinerU

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)
        return {"result": result}
    except Exception as e:
        return JSONResponse(status_code=500, content={"error": str(e)})