0x90e commited on
Commit
fcdd24e
·
1 Parent(s): f2475e0
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +27,7 @@ def inference(img, size, type):
27
  run_cmd(f"mkdir {OUTPUT_DIR}")
28
  img.save(INPUT_DIR + "1.jpg", "JPEG")
29
 
30
- if type == "Manga v2":
31
  run_cmd(f"python inference_manga_v2.py {img_in_path} {img_out_path}")
32
  else:
33
  run_cmd(f"python inference.py {img_in_path} {img_out_path} {type}")
@@ -47,7 +47,7 @@ def inference(img, size, type):
47
  return [img_out]
48
 
49
  input_image = gr.Image(type="pil", label="Input")
50
- upscale_type = gr.Radio(["Manga", "Anime", "General"], label="Select the type of picture you want to upscale:", value="Manga v2")
51
  upscale_size = gr.Radio(["x4", "x2"], label="Upscale by:", value="x4")
52
  output_image = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
53
 
 
27
  run_cmd(f"mkdir {OUTPUT_DIR}")
28
  img.save(INPUT_DIR + "1.jpg", "JPEG")
29
 
30
+ if type == "Manga":
31
  run_cmd(f"python inference_manga_v2.py {img_in_path} {img_out_path}")
32
  else:
33
  run_cmd(f"python inference.py {img_in_path} {img_out_path} {type}")
 
47
  return [img_out]
48
 
49
  input_image = gr.Image(type="pil", label="Input")
50
+ upscale_type = gr.Radio(["Manga", "Anime", "General"], label="Select the type of picture you want to upscale:", value="Manga")
51
  upscale_size = gr.Radio(["x4", "x2"], label="Upscale by:", value="x4")
52
  output_image = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
53