optipng

app.py

@@ -31,6 +31,7 @@ def run_cmd(command):
 
 run_cmd("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P .")
 run_cmd("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth -P .")
+run_cmd("apt-get update; apt-get install optipng")
 
 
 
@@ -51,7 +52,10 @@ def inference(img,mode):
         run_cmd("python inference_realesrgan.py -n RealESRGAN_x4plus -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
     else:
         run_cmd("python inference_realesrgan.py -n RealESRGAN_x4plus_anime_6B -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
-    return os.path.join(OUTPUT_DIR, "1_out.png")
+
+    image_filename = os.path.join(OUTPUT_DIR, "1_out.png")
+    run_cmd("optipng -o5 " + image_filename)
+    return image_filename