Update app.py
Browse files
app.py
CHANGED
|
@@ -33,9 +33,9 @@ def inference(img_path, Style, if_face=None):
|
|
| 33 |
det_face = True if if_face=="Yes" else False
|
| 34 |
output = AnimeGANv3_src.Convert(img, f, det_face)
|
| 35 |
result = cv2.resize(output, (img.shape[1],img.shape[0] ), interpolation = cv2.INTER_AREA)
|
| 36 |
-
|
| 37 |
-
cv2.imwrite(save_path, result[:, :, ::-1])
|
| 38 |
-
return result
|
| 39 |
except RuntimeError as error:
|
| 40 |
print('Error', error)
|
| 41 |
except Exception as error:
|
|
@@ -91,7 +91,7 @@ gr.Interface(
|
|
| 91 |
gr.inputs.Radio(['Yes', 'No'], type="value", default='No', label='Extract face'),
|
| 92 |
], [
|
| 93 |
gr.outputs.Image(type="numpy", label="Output (The whole image)"),
|
| 94 |
-
|
| 95 |
],
|
| 96 |
title=title,
|
| 97 |
description=description,
|
|
|
|
| 33 |
det_face = True if if_face=="Yes" else False
|
| 34 |
output = AnimeGANv3_src.Convert(img, f, det_face)
|
| 35 |
result = cv2.resize(output, (img.shape[1],img.shape[0] ), interpolation = cv2.INTER_AREA)
|
| 36 |
+
# save_path = f"output/out.{img_path.rsplit('.')[-1]}"
|
| 37 |
+
#cv2.imwrite(save_path, result[:, :, ::-1])
|
| 38 |
+
return result
|
| 39 |
except RuntimeError as error:
|
| 40 |
print('Error', error)
|
| 41 |
except Exception as error:
|
|
|
|
| 91 |
gr.inputs.Radio(['Yes', 'No'], type="value", default='No', label='Extract face'),
|
| 92 |
], [
|
| 93 |
gr.outputs.Image(type="numpy", label="Output (The whole image)"),
|
| 94 |
+
# gr.outputs.File(label="Download the output image")
|
| 95 |
],
|
| 96 |
title=title,
|
| 97 |
description=description,
|