Spaces:
Build error
Build error
File size: 1,100 Bytes
0ba02e5 b33e395 0ba02e5 469e3e7 0595a8e 17c6031 0595a8e 17c6031 0595a8e 17c6031 acff26c 0ba02e5 8c1034d 0ba02e5 8c1034d 17c6031 ac889d5 9230043 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import gradio as gr
import json
import requests
def test():
print('testing backend interface...')
url = 'http://81.70.158.155:5005/predict'
with open('./2680.jpg', 'rb') as f:
try:
r = requests.post(url, files = {'image' : f}, timeout=10)
ret = json.loads(r.text)
except Exception as e:
print('backend test failed')
return
# image level cls score
pred_cls_score = ret['cls_score']
print('--------', pred_cls_score)
print('backend test success')
def predict(img):
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
iface = gr.Interface(
predict,
inputs=gr.inputs.Image(label="Upload image to detect", type="filepath"),
outputs='image',
title="Forged? Or Not?",
)
test()
iface.launch()
|