Spaces:
Runtime error
Runtime error
File size: 2,922 Bytes
0e1f08b e3012f6 0e1f08b e3012f6 0e1f08b e3012f6 0e1f08b e3012f6 0e1f08b e3012f6 0e1f08b e3012f6 0e1f08b e3012f6 0e1f08b e3012f6 0e1f08b e3012f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import gradio as gr
from actors_matching.api import analyze_image, load_annoy_index
from pathlib import Path
annoy_index, actors_mapping = load_annoy_index()
def get_image_html(actor: dict):
url = actor["url"]
name = actor["name"]
imdb_url = f"https://www.imdb.com/name/{actor['nconst']}/"
return f'''
<div style="position: relative; text-align: center; color: white;">
<img src="{url}" alt="{name} matches the input image" style="height: 500px">
<div style="padding: 0.2em; position: absolute; bottom: 16px; left: 16px; background-color: #aacccccc; font-size: 2em;">
<p>{name}</p>
<p style="font-size:0.5em"><a href={imdb_url} target="_blank">Click to see on IMDb</></p>
</div>
</div>
'''
def no_faces_found_html():
return f"""<div>No faces found in the picture</div>"""
def get_best_matches(image, n_matches: int):
return analyze_image(image, annoy_index=annoy_index, n_matches=n_matches)
def find_matching_actors(input_img, title, n_matches: int = 10):
best_matches_list = get_best_matches(input_img, n_matches=n_matches)
# TODO: allow looping through characters
if best_matches_list:
best_matches = best_matches_list[0]
# TODO: Show how the initial image was parsed (ie: which person is displayed)
# Build htmls to display the result
output_htmls = []
for match in best_matches["matches"]:
actor = actors_mapping[match]
output_htmls.append(get_image_html(actor))
return output_htmls
# No matches
return [no_faces_found_html()]
iface = gr.Interface(
find_matching_actors,
title="Which actor or actress looks like you?",
description="""Who is the best person to play a movie about you? Upload a picture and find out!
Or maybe you'd like to know who would best interpret your favorite historical character?
Give it a shot or try one of the sample images below.\nPlease read below for more information on biases
and limitations of the tool!""",
article=Path("README.md").read_text(),
inputs=[
gr.inputs.Image(shape=(256, 256), label="Your image"),
gr.inputs.Textbox(label="Who's that?", placeholder="Optional, you can leave this blank"),
#gr.inputs.Slider(minimum=1, maximum=10, step=1, default=5, label="Number of matches"),
],
outputs=gr.outputs.Carousel(gr.outputs.HTML(), label="Matching actors & actresses"),
examples=[
["images/example_rb_ginsburg.jpg", "RB Ginsburg in 1977"],
["images/example_hannibal_barca.jpg", "Hannibal (the one with the elephants...)"],
["images/example_frederick_douglass.jpg", "Frederik Douglass"],
["images/example_leonardo_davinci.jpg", "Leonoardo da Vinci"],
["images/example_joan_of_arc.jpg", "Jeanne d'Arc"],
["images/example_sun_tzu.jpg", "Sun Tzu"],
]
)
iface.launch()
|