Improve output dir handling.

app.py

@@ -23,7 +23,7 @@ def inference(img):
     run_cmd("mkdir " + OUTPUT_DIR)
     img.save(INPUT_DIR + "1.jpg", "JPEG")
 
-    os.system("python test.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + OUTPUT_DIR)
+    os.system("python test.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
     return os.path.join(OUTPUT_DIR, "1_out.jpg")
 
 input_image = gr.inputs.Image(type="pil", label="Input")

test.py

@@ -30,4 +30,4 @@ img_LR = img_LR.to(device)
 
 output = model(img_LR).squeeze(dim=0).float().cpu().clamp_(0, 1).numpy()
 output = np.transpose(output, (1, 2, 0))
-cv2.imwrite(os.path.join(output_dir, "1_out.jpg"), output, [int(cv2.IMWRITE_JPEG_QUALITY), 90])
+cv2.imwrite(output_dir, output, [int(cv2.IMWRITE_JPEG_QUALITY), 90])