diff --git a/util/model_util.py b/util/model_util.py index 0e5aa2a..9d3324b 100644 --- a/util/model_util.py +++ b/util/model_util.py @@ -36,7 +36,7 @@ def ie_settlement(img_path, layout): response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)}) if response.status_code == 200: ie_result = response.json() - if ie_result.get(0): + if ie_result: return ie_result[0] return None @@ -53,7 +53,7 @@ def ie_settlement_text(text): response = requests.post(url, {'text': text}) if response.status_code == 200: ie_result = response.json() - if ie_result.get(0): + if ie_result: return ie_result[0] return None @@ -71,7 +71,7 @@ def ie_discharge(img_path, layout): response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)}) if response.status_code == 200: ie_result = response.json() - if ie_result.get(0): + if ie_result: return ie_result[0] return None @@ -88,7 +88,7 @@ def ie_discharge_text(text): response = requests.post(url, {'text': text}) if response.status_code == 200: ie_result = response.json() - if ie_result.get(0): + if ie_result: return ie_result[0] return None @@ -106,7 +106,7 @@ def ie_cost(img_path, layout): response = requests.post(url, {'img_path': img_path, 'layout': json.dumps(layout)}) if response.status_code == 200: ie_result = response.json() - if ie_result.get(0): + if ie_result: return ie_result[0] return None @@ -123,7 +123,7 @@ def ie_cost_text(text): response = requests.post(url, {'text': text}) if response.status_code == 200: ie_result = response.json() - if ie_result.get(0): + if ie_result: return ie_result[0] return None