Update app.py

app.py

@@ -12,7 +12,10 @@ from interpret import InterpretationPrompt
 MAX_PROMPT_TOKENS = 30
 
 ## info
-dataset_info = [{'name': 'Commonsense', 'hf_repo': 'tau/commonsense_qa', 'text_col': 'question'}]
+dataset_info = [{'name': 'Commonsense', 'hf_repo': 'tau/commonsense_qa', 'text_col': 'question'},
+                {'name': 'Fact Recall', 'hf_repo': 'azhx/counterfact-filtered-gptj6b', 'text_col': 'subject+predicate', 
+                 'filter': lambda x: x['label'] == 1},
+               ]
 
 
 
@@ -198,13 +201,15 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
         interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
 
     gr.Markdown('''
-    Here are some examples of prompts we can analyze their internal representations
+    Here are some examples of prompts we can analyze their internal representations:
     ''')
     
     for info in dataset_info:
         with gr.Tab(info['name']):
             num_examples = 10
             dataset = load_dataset(info['hf_repo'], split='train', streaming=True)
+            if 'filter' in info:
+                dataset = dataset.filter(info['filter'])
             dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
             dataset = [[row[info['text_col']]] for row in dataset]
             gr.Examples(dataset, [original_prompt_raw])