Improvements?

app.py

@@ -19,21 +19,30 @@ def inference(img, size, type):
     _id = randint(1, 10000)
     INPUT_DIR = "/tmp/input_image" + str(_id) + "/"
     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)
+    img_in_path = os.path.join(INPUT_DIR, "1.jpg")
+    img_out_path = os.path.join(OUTPUT_DIR, "1_out.jpg")
+    run_cmd(f"rm -rf {INPUT_DIR}")
+    run_cmd(f"rm -rf {OUTPUT_DIR}")
+    run_cmd(f"mkdir {INPUT_DIR}")
+    run_cmd(f"mkdir {OUTPUT_DIR}")
     img.save(INPUT_DIR + "1.jpg", "JPEG")
 
     if type == "Manga v2":
-        run_cmd("python inference_manga_v2.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
+        run_cmd(f"python inference_manga_v2.py {img_in_path} {img_out_path}")
     else:
-        run_cmd("python inference.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg") + " " + type)
+        run_cmd(f"python inference.py {img_in_path} {img_out_path} {type}")
 
-    img_out = Image.open(os.path.join(OUTPUT_DIR, "1_out.jpg"))
+    img_out = Image.open(img_out_path)
 
     if size == "x2":
-        img_out = img_out.resize((s//2 for s in img_out.size))
+        img_out = img_out.resize((img_out.width // 2, img_out.height // 2), resample=Image.BICUBIC)
+
+    img_out.save(img_out_path, optimize=True) # Add more optimizations
+    img_out = Image.open(img_out_path)
+
+    # Remove input and output image
+    run_cmd(f"rm -f {img_in_path}")
+    run_cmd(f"rm -f {img_out_path}")
 
     return [img_out]
 
@@ -48,4 +57,5 @@ demo = gr.Interface(
     outputs=[output_image]
 )
 
+demo.queue()
 demo.launch(debug=is_colab, share=is_colab, inline=is_colab)