From 35fd74f0a4283ab564a741351af2d462518b1f8a Mon Sep 17 00:00:00 2001 From: Kasisnu <5044120+kasisnu@users.noreply.github.com> Date: Wed, 1 May 2024 19:01:50 +0530 Subject: [PATCH] let preview server accept pngs _and_ jpegs as inputs --- workflows/preview_server/preview_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/workflows/preview_server/preview_server.py b/workflows/preview_server/preview_server.py index 3f06485..d0e0a30 100644 --- a/workflows/preview_server/preview_server.py +++ b/workflows/preview_server/preview_server.py @@ -191,7 +191,7 @@ async def preview_request(request_obj: Request, request: str = Form(...), input_ if 'video' in mime_type: input_filename_ext = ".mp4" elif 'image' in mime_type: - input_filename_ext = ".jpeg" + input_filename_ext = mimetypes.guess_extension(mime_type) else: raise HTTPException(status_code=400, detail="Unsupported file type")