Fixed error

run_llm.py

@@ -128,7 +128,6 @@ def para(m):
 def main(args=None):
 
     gid_list = selected_idx[args.start:args.end]
-    text_to_analyze = "Mr. Guber , by contrast , has been married to one woman for more than 20 years ."
 
 
     if 'gpt3' in args.model_path:
@@ -162,7 +161,7 @@ def main(args=None):
 
     if args.prompt == 1:
         for gid in tqdm(gid_list, desc='Query'):
-            text = text_to_analyze
+            text = ptb[gid]['text']
 
             for eid, ent in enumerate(ents):
                 os.makedirs(f'result/prompt1_qa/{args.model_path}/ptb/per_ent/{ent}', exist_ok=True)
@@ -217,7 +216,7 @@ def main(args=None):
 
     if args.prompt == 2:
         for gid in tqdm(gid_list, desc='Query'):
-            text = text_to_analyze
+            text = ptb[gid]['text']
 
             ## POS tagging
             # if os.path.exists(f'result/prompt2_instruction/pos_tagging/{args.model_path}/ptb/{gid}.txt'):
@@ -302,7 +301,7 @@ def main(args=None):
 
     if args.prompt == 3:
         for gid in tqdm(gid_list, desc='Query'):
-            text = text_to_analyze
+            text = ptb[gid]['text']
             tokens = ptb[gid]['tokens']
             poss = ptb[gid]['uni_poss']