Spaces:
Runtime error
Runtime error
Fixes.
Browse files
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
|
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
|
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]
|