a

ape/instance.py

@@ -106,8 +106,8 @@ def load_entity(file='./ape/data/entity_train.json'):
         input = i['text']
         output = []
         for j in i['labels']:
-            ##筛选局部实体类型，也可以拆分成单个实体类型
-            if j[1] in ['DRUG_DOSAGE', 'DRUG_TASTE', 'DRUG_EFFICACY']:
+            ##拆分成单实体任务类型
+            if j[1] in =='DRUG_EFFICACY':
                 output.append({'类型': j[1], '实体': j[-1]})
         output = json.dumps(output, ensure_ascii=False)
         data.append((input, output))

app.py

@@ -21,7 +21,7 @@ with gr.Blocks(title="Automatic Prompt Engineer", theme=gr.themes.Glass()) as de
                 n_eval = gr.Slider(label="评估样本数", minimum=1, maximum=30, step=5, value=20)
 
         with gr.Column(scale=3):
-            gr.Markdown("## 第二步：加载数据")
+            gr.Markdown("## 第二步：加载数据(选任务或上传数据)")
             with gr.Tab("选择数据"):
                 with gr.Row():
                     file = gr.File(label='上传txt文件，input[空格]output[换行]')