diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index bec0b21..c00fc0a 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -395,11 +395,12 @@ 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), + task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1), Taskflow("information_extraction", schema=DISCHARGE_RECORD_SCHEMA, model="uie-x-base", - task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS), + task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS, + device_id=-1), Taskflow("information_extraction", schema=COST_LIST_SCHEMA, model="uie-x-base", - task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS), + task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1), ] for phhd in phhds: pk_phhd = phhd.pk_phhd