test5

app.py

@@ -50,8 +50,8 @@ for i, j in zip(ents, ents_prompt):
     print(i, j)
 
 model_mapping = {
-    #'gpt3.5': 'gpt2',
-    'vicuna-7b': 'lmsys/vicuna-7b-v1.3',
+    'gpt3.5': 'gpt2',
+    #'vicuna-7b': 'lmsys/vicuna-7b-v1.3',
     #'vicuna-13b': 'lmsys/vicuna-13b-v1.3',
     #'vicuna-33b': 'lmsys/vicuna-33b-v1.3',
     #'fastchat-t5': 'lmsys/fastchat-t5-3b-v1.0',
@@ -116,7 +116,7 @@ def process_text(model_name, task, text):
     for gid in tqdm(gid_list, desc='Query'):
             text = ptb[gid]['text']
     
-    if model_name == 'gpt2': 
+    if model_name == 'gpt3.5': 
         if task == 'POS':
             strategy1_format = template_all.format(text)
             strategy2_format = prompt2_pos.format(text)
@@ -126,9 +126,9 @@ def process_text(model_name, task, text):
             result2 = gpt_pipeline(strategy2_format)
             result3 = gpt_pipeline(strategy3_format)
 
-            generated_text1 = result1[0]['generated_text']
-            generated_text2 = result2[0]['generated_text']
-            generated_text3 = result3[0]['generated_text']
+            generated_text1 = result1[0]['sequence']
+            generated_text2 = result2[0]['sequence']
+            generated_text3 = result3[0]['sequence']
 
             return (generated_text1, generated_text2, generated_text3)
         # elif task == 'Chunking':