Update app.py

app.py

@@ -35,18 +35,18 @@ def inference(img,mode):
     OUTPUT_DIR = "/tmp/output_image" + str(_id) + "/"
     run_cmd("rm -rf " + INPUT_DIR)
     run_cmd("rm -rf " + OUTPUT_DIR)
-    run_cmd("mkdir " + INPUT_DIR)
-    run_cmd("mkdir " + OUTPUT_DIR)
+    run_cmd("mkdir -p" + INPUT_DIR)
+    run_cmd("mkdir -p" + OUTPUT_DIR)
     basewidth = 256
     wpercent = (basewidth/float(img.size[0]))
     hsize = int((float(img.size[1])*float(wpercent)))
     #img = img.resize((basewidth,hsize), Image.LANCZOS)
-    img.save(INPUT_DIR + "1.jpg", "JPEG")
+    img.save(INPUT_DIR + "1.ppg", "PNG")
     if mode == "base":
         run_cmd("python inference_realesrgan.py -n RealESRGAN_x4plus -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
     else:
         os.system("python inference_realesrgan.py -n RealESRGAN_x4plus_anime_6B -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
-    return os.path.join(OUTPUT_DIR, "1_out.jpg")
+    return os.path.join(OUTPUT_DIR, "1_out.png")