Spaces:
Runtime error
Runtime error
fix image resizing and update article info
Browse files
app.py
CHANGED
@@ -1,8 +1,10 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
|
|
2 |
from actors_matching.api import analyze_image, load_annoy_index
|
3 |
from pathlib import Path
|
4 |
|
5 |
-
|
6 |
annoy_index, actors_mapping = load_annoy_index()
|
7 |
|
8 |
|
@@ -29,8 +31,22 @@ def get_best_matches(image, n_matches: int):
|
|
29 |
return analyze_image(image, annoy_index=annoy_index, n_matches=n_matches)
|
30 |
|
31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
def find_matching_actors(input_img, title, n_matches: int = 10):
|
33 |
-
|
|
|
34 |
|
35 |
# TODO: allow looping through characters
|
36 |
if best_matches_list:
|
@@ -55,9 +71,13 @@ iface = gr.Interface(
|
|
55 |
title="Which actor or actress looks like you?",
|
56 |
description="""Who is the best person to play a movie about you? Upload a picture and find out!
|
57 |
Or maybe you'd like to know who would best interpret your favorite historical character?
|
58 |
-
Give it a shot or try one of the sample images below
|
|
|
|
|
|
|
|
|
59 |
and limitations of the tool!""",
|
60 |
-
article=
|
61 |
inputs=[
|
62 |
gr.inputs.Image(shape=None, label="Your image"),
|
63 |
gr.inputs.Textbox(
|
|
|
1 |
import gradio as gr
|
2 |
+
import PIL
|
3 |
+
import numpy as np
|
4 |
+
import re
|
5 |
from actors_matching.api import analyze_image, load_annoy_index
|
6 |
from pathlib import Path
|
7 |
|
|
|
8 |
annoy_index, actors_mapping = load_annoy_index()
|
9 |
|
10 |
|
|
|
31 |
return analyze_image(image, annoy_index=annoy_index, n_matches=n_matches)
|
32 |
|
33 |
|
34 |
+
def resize_image_keep_ratio(input_image: np.array, size: tuple):
|
35 |
+
resized_image = PIL.Image.fromarray(input_image)
|
36 |
+
resized_image.thumbnail(size, PIL.Image.ANTIALIAS)
|
37 |
+
return np.array(resized_image)
|
38 |
+
|
39 |
+
|
40 |
+
def get_article_text():
|
41 |
+
article = Path("README.md").read_text()
|
42 |
+
# Remove the HuggingFace Space app information from the README
|
43 |
+
article = re.sub(r"^---.+---\s+", "", article, flags=re.MULTILINE + re.DOTALL)
|
44 |
+
return article
|
45 |
+
|
46 |
+
|
47 |
def find_matching_actors(input_img, title, n_matches: int = 10):
|
48 |
+
resized_image = resize_image_keep_ratio(input_img, (512, 512))
|
49 |
+
best_matches_list = get_best_matches(resized_image, n_matches=n_matches)
|
50 |
|
51 |
# TODO: allow looping through characters
|
52 |
if best_matches_list:
|
|
|
71 |
title="Which actor or actress looks like you?",
|
72 |
description="""Who is the best person to play a movie about you? Upload a picture and find out!
|
73 |
Or maybe you'd like to know who would best interpret your favorite historical character?
|
74 |
+
Give it a shot or try one of the sample images below.
|
75 |
+
|
76 |
+
Built with ❤️ using great open-source libraries such as dlib, face_recognition and Annoy.
|
77 |
+
|
78 |
+
Please read below for more information on biases
|
79 |
and limitations of the tool!""",
|
80 |
+
article=get_article_text(),
|
81 |
inputs=[
|
82 |
gr.inputs.Image(shape=None, label="Your image"),
|
83 |
gr.inputs.Textbox(
|