diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index c00fc0a..bec0b21 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -395,12 +395,11 @@ def main(): if phhds: ie_task_flows = [ Taskflow("information_extraction", schema=SETTLEMENT_LIST_SCHEMA, model="uie-x-base", - task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1), + task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS), Taskflow("information_extraction", schema=DISCHARGE_RECORD_SCHEMA, model="uie-x-base", - task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS, - device_id=-1), + task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS), Taskflow("information_extraction", schema=COST_LIST_SCHEMA, model="uie-x-base", - task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1), + task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS), ] for phhd in phhds: pk_phhd = phhd.pk_phhd