Spaces:
Running
Running
| pip install insightface | |
| pip install onnxruntime | |
| import gradio as gr | |
| import numpy as np | |
| import tensorflow as tf | |
| import cv2 | |
| import numpy as np | |
| import os | |
| import glob | |
| import cv2 | |
| import matplotlib.pyplot as plt | |
| import insightface | |
| from insightface.app import FaceAnalysis | |
| from insightface.data import get_image as ins_get_image | |
| # Load your trained model | |
| #model = tf.keras.models.load_model('path_to_your_model.h5') | |
| def predict_gender(image): | |
| # Convert image to format expected by your model & preprocess | |
| img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) | |
| img = cv2.resize(img, (224, 224)) # Example size | |
| img = img / 255.0 # Normalizing | |
| img = np.expand_dims(img, axis=0) | |
| prediction = model.predict(img) | |
| # Assuming binary classification with a single output neuron | |
| return "Male" if prediction[0] < 0.5 else "Female" | |
| def predict(video_in, image_in_video, image_in_img): | |
| if video_in == None and image_in_video == None and image_in_img == None: | |
| raise gr.Error("Please upload a video or image.") | |
| if image_in_video or image_in_img: | |
| print("image", image_in_video, image_in_img) | |
| image = image_in_video or image_in_img | |
| return image | |
| return video_in | |
| def toggle(choice): | |
| if choice == "webcam": | |
| return gr.update(visible=True, value=None), gr.update(visible=False, value=None) | |
| else: | |
| return gr.update(visible=False, value=None), gr.update(visible=True, value=None) | |
| with gr.Blocks() as blocks: | |
| gr.Markdown("### Video or Image? WebCam or Upload?""") | |
| with gr.Tab("Video") as tab: | |
| with gr.Row(): | |
| with gr.Column(): | |
| video_or_file_opt = gr.Radio(["webcam", "upload"], value="webcam", | |
| label="How would you like to upload your video?") | |
| video_in = gr.Video(source="webcam", include_audio=False) | |
| video_or_file_opt.change(fn=lambda s: gr.update(source=s, value=None), inputs=video_or_file_opt, | |
| outputs=video_in, queue=False, show_progress=False) | |
| with gr.Column(): | |
| video_out = gr.Video() | |
| run_btn = gr.Button("Run") | |
| run_btn.click(fn=predict, inputs=[video_in], outputs=[video_out]) | |
| gr.Examples(fn=predict, examples=[], inputs=[ | |
| video_in], outputs=[video_out]) | |
| with gr.Tab("Image"): | |
| with gr.Row(): | |
| with gr.Column(): | |
| image_or_file_opt = gr.Radio(["webcam", "file"], value="webcam", | |
| label="How would you like to upload your image?") | |
| image_in_video = gr.Image(source="webcam", type="filepath") | |
| image_in_img = gr.Image( | |
| source="upload", visible=False, type="filepath") | |
| image_or_file_opt.change(fn=toggle, inputs=[image_or_file_opt], | |
| outputs=[image_in_video, image_in_img], queue=False, show_progress=False) | |
| with gr.Column(): | |
| image_out = gr.Image() | |
| run_btn = gr.Button("Run") | |
| run_btn.click(fn=predict, inputs=[ | |
| image_in_img, image_in_video], outputs=[image_out]) | |
| gr.Examples(fn=predict, examples=[], inputs=[ | |
| image_in_img, image_in_video], outputs=[image_out]) | |
| blocks.queue() | |
| blocks.launch() |