vjain commited on
Commit
3ea1e2b
·
1 Parent(s): e37d22f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -36,11 +36,16 @@ AP_statistics = load_dataset('vjain/AP_statistics')
36
  df4 = pd.DataFrame(AP_statistics['train'])
37
  df4["similarity"] = 0
38
 
 
 
 
 
39
  dataframes = {
40
  "AP_Bio": df1,
41
  "AP_Physics": df2,
42
  "Personality" : df3,
43
- "AP_statistics": df4
 
44
  }
45
 
46
  #df = pd.read_csv("TA_embeddings.csv")
@@ -83,7 +88,7 @@ def reply(input, dataset_name):
83
 
84
  csv_dropdown = gr.inputs.Dropdown(
85
  label="Select the Book",
86
- choices=["AP_Bio", "AP_Physics","Personality","AP_statistics"],
87
  default="AP_Bio"
88
 
89
  )
 
36
  df4 = pd.DataFrame(AP_statistics['train'])
37
  df4["similarity"] = 0
38
 
39
+ Tax_embeddings = load_dataset("vjain/tax_embeddings")
40
+ df5 = pd.dataframe(Tax_embeddings['train'])
41
+ df5["similarity"] = 0
42
+
43
  dataframes = {
44
  "AP_Bio": df1,
45
  "AP_Physics": df2,
46
  "Personality" : df3,
47
+ "AP_statistics": df4,
48
+ "Tax_embeddings": df5
49
  }
50
 
51
  #df = pd.read_csv("TA_embeddings.csv")
 
88
 
89
  csv_dropdown = gr.inputs.Dropdown(
90
  label="Select the Book",
91
+ choices=["AP_Bio", "AP_Physics","Personality","AP_statistics","Tax_embeddings"],
92
  default="AP_Bio"
93
 
94
  )