File size: 633 Bytes
0ba02e5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8c1034d
0ba02e5
 
 
 
8c1034d
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
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
  
        

iface = gr.Interface(
    predict,
    inputs=gr.inputs.Image(label="Upload image to detect", type="filepath"),
    outputs='image',
    title="Forged? Or Not?",
)
iface.launch(share=True)

print('--------------->')