init

app.py

@@ -16,6 +16,7 @@ import PIL.Image
 
 from io import BytesIO
 from fastai.vision import *
+from fastai.utils.mem import *
 from fastai.vision import load_learner
 
 from core import FeatureLoss
@@ -65,12 +66,13 @@ def run(
 
     img = PIL.Image.open(image.name)
     img_t = T.ToTensor()(img)
-    img_fast = PIL.Image(img_t)
+    img_fast = Image(img_t)
 
     p, img_hr, b = learn.predict(img_fast)
-    r = PIL.Image(img_hr)
 
-    return r
+    r = np.uint8(np.clip(image2np(img_hr), 0, 1) * 255)
+
+    return PIL.Image(r)
 
 learn = None
 def main():