research14 commited on
Commit
9187ce2
·
1 Parent(s): 795f139
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def process_text(model_name, task, text):
116
  for gid in tqdm(gid_list, desc='Query'):
117
  text = ptb[gid]['text']
118
 
119
- if model_name == 'vicuna-7b':
120
  if task == 'POS':
121
  strategy1_format = template_all.format(text)
122
  strategy2_format = prompt2_pos.format(text)
 
116
  for gid in tqdm(gid_list, desc='Query'):
117
  text = ptb[gid]['text']
118
 
119
+ if model_name == 'gpt2':
120
  if task == 'POS':
121
  strategy1_format = template_all.format(text)
122
  strategy2_format = prompt2_pos.format(text)