Sa-m's picture
Update app.py
a58dbb4
raw
history blame contribute delete
718 Bytes
"""
Run a rest API exposing the yolov5s object detection model
"""
import gradio as gr
import requests
def detect(inp):
filename=inp
bs64Data=gr.processing_utils.encode_file_to_base64(filename)
r = requests.post(url='https://hf.space/gradioiframe/Sa-m/Political-Party-Symbol-Detector-V1/+/api/predict/',
json={"data": [bs64Data]})
res= r.json()
res['data']='hidden since it is a large base64 file'
return res
inp = gr.inputs.Image(type='filepath', label="Input Image")
output=gr.outputs.JSON(label='Response')
io=gr.Interface(fn=detect, inputs=inp, outputs=output, title='Party Symbol Detector API',examples=['8.jpg','24.jpg'])
io.launch(debug=True,share=False)