gg

.gitignore

@@ -0,0 +1 @@
+/gg

app.py

@@ -1,14 +1,13 @@
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import StreamingResponse
-from ./image_enhancer import EnhancementMethod, Enhancer
-from vidoe_enhancer import VideoEnhancer
+from image_enhancer import EnhancementMethod, Enhancer
+from video_enhancer import VideoEnhancer
 from pydantic import BaseModel
 from PIL import Image
 from io import BytesIO
 import base64
-import numpy as np
 import magic
-
+from typing import List
 class EnhancementRequest(BaseModel):
     method: EnhancementMethod = EnhancementMethod.gfpgan
     background_enhancement: bool = True
@@ -46,6 +45,7 @@ async def enhance_image(
         img_byte_arr = BytesIO()
         enhanced_image.save(img_byte_arr, format='PNG')
         img_byte_arr.seek(0)
+        print(original_resolution, enhanced_resolution)
         
         return StreamingResponse(img_byte_arr, media_type="image/png")
 

video_enhancer.py

@@ -3,6 +3,7 @@ import numpy as np
 import torch
 import io
 import asyncio
+from fastapi.responses import StreamingResponse
 from basicsr.archs.rrdbnet_arch import RRDBNet
 from realesrgan import RealESRGANer
 from huggingface_hub import hf_hub_download