Spaces:
Running
Running
Update
Browse files
app.py
CHANGED
|
@@ -14,6 +14,11 @@ import gradio as gr
|
|
| 14 |
import huggingface_hub
|
| 15 |
import numpy as np
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
TOKEN = os.environ['TOKEN']
|
| 18 |
|
| 19 |
|
|
@@ -63,8 +68,7 @@ def detect(image, detector: cv2.CascadeClassifier) -> np.ndarray:
|
|
| 63 |
res = image.copy()
|
| 64 |
for x, y, w, h in preds:
|
| 65 |
cv2.rectangle(res, (x, y), (x + w, y + h), (0, 255, 0), 2)
|
| 66 |
-
res
|
| 67 |
-
return res
|
| 68 |
|
| 69 |
|
| 70 |
def main():
|
|
@@ -79,20 +83,15 @@ def main():
|
|
| 79 |
func = functools.partial(detect, detector=detector)
|
| 80 |
func = functools.update_wrapper(func, detect)
|
| 81 |
|
| 82 |
-
repo_url = 'https://github.com/nagadomi/lbpcascade_animeface'
|
| 83 |
-
title = 'nagadomi/lbpcascade_animeface'
|
| 84 |
-
description = f'A demo for {repo_url}'
|
| 85 |
-
article = None
|
| 86 |
-
|
| 87 |
gr.Interface(
|
| 88 |
func,
|
| 89 |
gr.inputs.Image(type='file', label='Input'),
|
| 90 |
-
gr.outputs.Image(label='Output'),
|
| 91 |
-
theme=args.theme,
|
| 92 |
-
title=title,
|
| 93 |
-
description=description,
|
| 94 |
-
article=article,
|
| 95 |
examples=examples,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 96 |
allow_screenshot=args.allow_screenshot,
|
| 97 |
allow_flagging=args.allow_flagging,
|
| 98 |
live=args.live,
|
|
|
|
| 14 |
import huggingface_hub
|
| 15 |
import numpy as np
|
| 16 |
|
| 17 |
+
ORIGINAL_REPO_URL = 'https://github.com/nagadomi/lbpcascade_animeface'
|
| 18 |
+
TITLE = 'nagadomi/lbpcascade_animeface'
|
| 19 |
+
DESCRIPTION = f'A demo for {ORIGINAL_REPO_URL}'
|
| 20 |
+
ARTICLE = None
|
| 21 |
+
|
| 22 |
TOKEN = os.environ['TOKEN']
|
| 23 |
|
| 24 |
|
|
|
|
| 68 |
res = image.copy()
|
| 69 |
for x, y, w, h in preds:
|
| 70 |
cv2.rectangle(res, (x, y), (x + w, y + h), (0, 255, 0), 2)
|
| 71 |
+
return res[:, :, ::-1]
|
|
|
|
| 72 |
|
| 73 |
|
| 74 |
def main():
|
|
|
|
| 83 |
func = functools.partial(detect, detector=detector)
|
| 84 |
func = functools.update_wrapper(func, detect)
|
| 85 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 86 |
gr.Interface(
|
| 87 |
func,
|
| 88 |
gr.inputs.Image(type='file', label='Input'),
|
| 89 |
+
gr.outputs.Image(type='numpy', label='Output'),
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
examples=examples,
|
| 91 |
+
title=TITLE,
|
| 92 |
+
description=DESCRIPTION,
|
| 93 |
+
article=ARTICLE,
|
| 94 |
+
theme=args.theme,
|
| 95 |
allow_screenshot=args.allow_screenshot,
|
| 96 |
allow_flagging=args.allow_flagging,
|
| 97 |
live=args.live,
|