yangwu
update app.py
f4c257e
raw
history blame
604 Bytes
import gradio as gr
import requests
def predict(name):
url = 'http://81.70.158.155:5005/predict'
with open(img, 'rb') as f:
r = requests.post(url, files = {'image' : f})
ret = json.loads(r.text)
# image level cls score
pred_cls_score = ret['cls_score']
pred_seg = json.loads(ret['result'])
pred_seg = np.array(pred_seg, dtype=np.uint8)
return pred_seg
gr.Interface(
predict,
inputs=gr.inputs.Image(label="Upload image to detect", type="filepath"),
outputs='image',
title="Forged? Or Not?",
).launch()
#).launch(share=True)