moazx commited on
Commit
ea8f6cb
·
1 Parent(s): b857db3

Fix FastAPI response annotation for /output route

Browse files
Files changed (1) hide show
  1. hf_app.py +3 -3
hf_app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  import shutil
7
 
8
  import torch
9
- from fastapi import FastAPI, File, Form, Request, UploadFile
10
  from fastapi.responses import FileResponse, HTMLResponse, JSONResponse
11
  from fastapi.staticfiles import StaticFiles
12
  from fastapi.templating import Jinja2Templates
@@ -292,8 +292,8 @@ async def pdf_details(pdf_stem: str) -> JSONResponse:
292
  )
293
 
294
 
295
- @app.get("/output/{filename:path}")
296
- async def output_file(filename: str) -> FileResponse | JSONResponse:
297
  """Serve output files (PDFs, images, markdown)."""
298
 
299
  output_root = OUTPUT_FOLDER.resolve()
 
6
  import shutil
7
 
8
  import torch
9
+ from fastapi import FastAPI, File, Form, Request, Response, UploadFile
10
  from fastapi.responses import FileResponse, HTMLResponse, JSONResponse
11
  from fastapi.staticfiles import StaticFiles
12
  from fastapi.templating import Jinja2Templates
 
292
  )
293
 
294
 
295
+ @app.get("/output/{filename:path}", response_model=None)
296
+ async def output_file(filename: str):
297
  """Serve output files (PDFs, images, markdown)."""
298
 
299
  output_root = OUTPUT_FOLDER.resolve()