Spaces:
Runtime error
Runtime error
| # Temporary fix to install other requirements requirements here | |
| # Pip needs to be ugraded on Hugging Face spaces before installing the library | |
| import os | |
| os.system("pip install --upgrade pip") | |
| os.system("pip install umap-learn bechdelai==0.0.1a2") | |
| import gradio as gr | |
| import os | |
| from bechdelai.image.face_detection import FacesDetector | |
| from bechdelai.image.gender_detection import GenderDetector | |
| from bechdelai.image.vilt import ViLT | |
| from bechdelai.image.img import Img | |
| fd = FacesDetector() | |
| gd = GenderDetector() | |
| examples_path = "static/examples" | |
| examples_images = [os.path.join(examples_path,x) for x in os.listdir(examples_path)] | |
| def analyze_genders(img): | |
| rois,faces = fd.detect(img,method = "retinaface",padding = 20) | |
| probas = gd.predict(faces) | |
| results,metrics = gd.analyze_probas(probas) | |
| img_with_faces = fd.show_faces_on_image(img,rois,width = 3,genders = probas["gender"]) | |
| new_metrics = {"Space occupied by women":metrics["women_area"],"Space occupied by men":metrics["men_area"]} | |
| return new_metrics, img_with_faces | |
| demo = gr.Interface( | |
| fn = analyze_genders, | |
| inputs = gr.Image(), | |
| outputs = [ | |
| gr.outputs.Label(num_top_classes=3), | |
| "image", | |
| ], | |
| examples=examples_images, | |
| title="BechdelAI Tool - Image Analysis", | |
| description="BechdelAI automates the Bechdel test and study of women under-representation in cinema and media with AI", | |
| article="For more information visit https://github.com/dataforgoodfr/bechdelai" | |
| ) | |
| demo.launch() |