0x90e commited on
Commit
9aa7f1e
·
1 Parent(s): 97c239f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,14 +26,14 @@ def inference(img, size, type):
26
  run_cmd("mkdir " + OUTPUT_DIR)
27
  img.save(INPUT_DIR + "1.jpg", "JPEG")
28
 
29
- if type is "Manga v2":
30
  run_cmd("python inference_manga_v2.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
31
  else:
32
  run_cmd("python inference.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg") + " " + type)
33
 
34
  img_out = Image.open(os.path.join(OUTPUT_DIR, "1_out.jpg"))
35
 
36
- if size is "x2":
37
  img_out = img_out.resize((s//2 for s in img_out.size))
38
 
39
  return [img_out]
 
26
  run_cmd("mkdir " + OUTPUT_DIR)
27
  img.save(INPUT_DIR + "1.jpg", "JPEG")
28
 
29
+ if type == "Manga v2":
30
  run_cmd("python inference_manga_v2.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
31
  else:
32
  run_cmd("python inference.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg") + " " + type)
33
 
34
  img_out = Image.open(os.path.join(OUTPUT_DIR, "1_out.jpg"))
35
 
36
+ if size == "x2":
37
  img_out = img_out.resize((s//2 for s in img_out.size))
38
 
39
  return [img_out]