Upload 2 files

fn.py

@@ -1,4 +1,5 @@
 import os
+import io
 import torch
 import numpy as np
 from PIL import Image

main.py

@@ -3,6 +3,7 @@ import sys
 import time
 import signal
 import io
+from PIL import Image
 
 from fastapi import FastAPI, Request, status, Form, UploadFile
 from fastapi.staticfiles import StaticFiles
@@ -31,13 +32,10 @@ fn.load_model()
 
 @app.post("/katanuki")
 async def katanuki_image(file: UploadFile = Form(...)):
-    try:
-        file_content = await file.read()
-        file_stream = io.BytesIO(file_content)
+    file_content = await file.read()
+    file_stream = io.BytesIO(file_content)
 
-        dst_image = fn.animeseg(file_stream)
-        binary = fn.pil_to_webp(dst_image)
+    dst_image = fn.animeseg(Image.open(file_stream))
+    binary = fn.pil_to_webp(Image.fromarray(dst_image))
 
-        return Response(content=binary, media_type="image/webp")
-    except Exception as e:
-        return {"error": str(e)}
+    return Response(content=binary, media_type="image/webp")