Integrate example topics and add fse prompt

app.py

@@ -8,7 +8,7 @@ import pandas as pd
 DATASETS_PATH = 'datasets'
 
 models = {
-  'mistral': ReplicateModel('mistralai/mistral-7b-instruct-v0.1:83b6a56e7c828e667f21fd596c338fd4f0039b46bcfa18d973e8e70e455fda70'),
+  'mistral_instruct': ReplicateModel('mistralai/mistral-7b-instruct-v0.1:83b6a56e7c828e667f21fd596c338fd4f0039b46bcfa18d973e8e70e455fda70'),
 }
 
 prompts = {
@@ -16,6 +16,26 @@ prompts = {
     '''
     I have topic that is described by the following keywords: [KEYWORDS]
 
+    Based on the information above, extract a short topic label in the following format:
+    topic: <topic label>
+    ''',
+
+  'few_shot_examples':
+    '''
+    I have a topic that is described by the following keywords: [KEYWORDS]
+
+    Example 1:
+    Keywords: apple,fruit,healthy,snack,red,orchard
+    Topic label: Healthy Fruit Snacks
+
+    Example 2:
+    Keywords: computer,technology,silicon,programming,internet,hardware
+    Topic label: Computer Technology
+
+    Example 3:
+    Keywords: democracy,government,elections,vote,political,representation
+    Topic label: Democratic Governance
+
     Based on the information above, extract a short topic label in the following format:
     topic: <topic label>
     '''
@@ -23,7 +43,7 @@ prompts = {
 }
 
 topicsets = {
-  'lda_poe_topics': os.path.join(DATASETS_PATH, 'lda_poe_topics.csv'),
+  'example_topics': os.path.join(DATASETS_PATH, 'topics.csv'),
 }
 
 @st.cache_data(show_spinner=False)
@@ -61,10 +81,12 @@ if sel_model_name:
 
 sel_dataset_name = col1.selectbox('Select a dataset', topicsets.keys(), index=None)
 if sel_dataset_name:
-  sel_dataset = pd.read_csv(topicsets[sel_dataset_name], header=None)
+  sel_dataset = pd.read_csv(topicsets[sel_dataset_name])
+  sel_dataset.drop(columns=['topic_id', 'domain'], inplace=True)
+
   col1.dataframe(sel_dataset)
 
-  sel_row_index = col1.selectbox('Select a row', sel_dataset.index)
+  sel_row_index = col1.selectbox('Select a topic', sel_dataset.index)
 
 sel_prompt = col2.selectbox('Select a prompt', prompts.keys())
 if sel_prompt != 'custom_prompt':