diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index 51428ac..1b0e0d2 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -38,6 +38,8 @@ def information_extraction(ie, phrecs): if pic_path: docs.append({"doc": pic_path}) doc_phrecs.append(phrec) + if not docs: + return result ie_results = ie(docs) now = get_default_datetime() diff --git a/visual_model_test/visual_model_test.py b/visual_model_test/visual_model_test.py index 39b0729..80c0e59 100644 --- a/visual_model_test/visual_model_test.py +++ b/visual_model_test/visual_model_test.py @@ -13,8 +13,8 @@ from ucloud import ucloud def write_visual_result(image, layout=None, result=None): - img = re.split(r'[\\/]', image)[-1] - img = img.split("?")[0] + img = image.split("?")[0] + img = re.split(r'[\\/]', img)[-1] img_name = "" img_type = "jpg" last_dot_index = img.rfind(".")