From 5c0fc0f8195aebbdda9a9ea743b558895876ccd7 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Thu, 10 Oct 2024 14:45:39 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3list=E5=80=BC=E7=9A=84?=
 =?UTF-8?q?=E8=8E=B7=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 util/model_util.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

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