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)