dar-tau commited on
Commit
04aee9f
1 Parent(s): 5ae57a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -15,6 +15,8 @@ MAX_PROMPT_TOKENS = 60
15
  dataset_info = [{'name': 'Commonsense', 'hf_repo': 'tau/commonsense_qa', 'text_col': 'question'},
16
  {'name': 'Factual Recall', 'hf_repo': 'azhx/counterfact-filtered-gptj6b', 'text_col': 'subject+predicate',
17
  'filter': lambda x: x['label'] == 1},
 
 
18
  ]
19
 
20
 
 
15
  dataset_info = [{'name': 'Commonsense', 'hf_repo': 'tau/commonsense_qa', 'text_col': 'question'},
16
  {'name': 'Factual Recall', 'hf_repo': 'azhx/counterfact-filtered-gptj6b', 'text_col': 'subject+predicate',
17
  'filter': lambda x: x['label'] == 1},
18
+ {'name': 'Physical Understanding', 'hf_repo': 'piqa', 'text_col': 'goal'},
19
+ {'name': 'Social Reasoning', 'hf_repo': 'ProlificAI/social-reasoning-rlhf', 'text_col': 'question'}
20
  ]
21
 
22