Update app.py

app.py

@@ -29,9 +29,14 @@ AP_Physics = load_dataset('vjain/AP_physics_embeddings')
 df2 = pd.DataFrame(AP_Physics['train'])
 df2["similarity"] = 0
 
+Personality = load_dataset('vjain/Personality_em')
+df3 = pd.DataFrame(Personality['train'])
+df3["similarity"] = 0
+
 dataframes = {
     "AP_Bio": df1,
-    "AP_Physics": df2
+    "AP_Physics": df2,
+    "Personality" : df3
 }
 
 #df = pd.read_csv("TA_embeddings.csv")
@@ -74,7 +79,7 @@ def reply(input, dataset_name):
 
 csv_dropdown = gr.inputs.Dropdown(
     label="Select the Book",
-    choices=["AP_Bio", "AP_Physics"],
+    choices=["AP_Bio", "AP_Physics","Personality"],
     default="AP_Bio"
             
 )