Spaces:
Build error
Build error
import gradio as gr | |
import json | |
import requests | |
import os | |
def test(): | |
os.system('ping -c 5 www.baidu.com ') | |
os.system('ping -c 5 www.google.com ') | |
os.system('ping -c 5 81.70.158.155 ') | |
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: ', str(e)) | |
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() | |