|
|
import sys |
|
|
import os |
|
|
import io |
|
|
import base64 |
|
|
import json |
|
|
import gradio as gr |
|
|
import requests |
|
|
|
|
|
|
|
|
def face_recognition_on_file(file1, file2): |
|
|
url = "http://93.127.215.33:8080/face_recognition" |
|
|
try: |
|
|
files = {'file1': open(file1, 'rb'), 'file2': open(file2, 'rb')} |
|
|
|
|
|
r = requests.post(url=url, files=files) |
|
|
r.raise_for_status() |
|
|
except requests.RequestException as e: |
|
|
raise gr.Error(f"Error occurred: {str(e)}") |
|
|
except IOError: |
|
|
raise gr.Error("Please select valid image files!") |
|
|
|
|
|
try: |
|
|
response = r.json() |
|
|
print(response) |
|
|
return response |
|
|
except json.JSONDecodeError: |
|
|
raise gr.Error("Invalid response from server") |
|
|
|
|
|
|
|
|
with gr.Blocks() as demo: |
|
|
gr.Markdown( |
|
|
""" |
|
|
# FacePlugin Online Demo |
|
|
|
|
|
""" |
|
|
) |
|
|
|
|
|
with gr.TabItem("Face Recognition"): |
|
|
with gr.Row(): |
|
|
with gr.Column(): |
|
|
first_input = gr.Image(type='filepath') |
|
|
gr.Examples(['images/rec_5.jpg', 'images/rec_1.jpg', 'images/9.png', 'images/rec_3.jpg'], |
|
|
inputs=first_input) |
|
|
start_button = gr.Button("Run") |
|
|
with gr.Column(): |
|
|
second_input = gr.Image(type='filepath') |
|
|
gr.Examples(['images/rec_6.jpg', 'images/rec_2.jpg', 'images/10.jpg', 'images/rec_4.jpg'], |
|
|
inputs=second_input) |
|
|
|
|
|
with gr.Column(): |
|
|
app_output = [gr.JSON()] |
|
|
|
|
|
start_button.click(face_recognition_on_file, inputs=[first_input, second_input], outputs=app_output) |
|
|
demo.queue().launch(share=True) |
|
|
|
