File size: 604 Bytes
0ba02e5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f4c257e
 
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
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)