Update app.py

app.py

@@ -7,12 +7,12 @@ from sentencex import segment
 
 model = SentenceTransformer("Corran/SciGenAllMiniLM")
 
-train = load_dataset("Corran/SciGenColbertTriplets")['train']
+rf = load_dataset("Corran/RhetoricFunctionsList")['train']['rhetoric_function']
 
-rf = sorted(list(set(train['query'])))
+rf = list(rf)
 rf_emb = model.encode(rf)
 index = Index(ndim=rf_emb[0].size)
-index.add(range(len(classes)), classes_emb)
+index.add(range(len(rf)), rf_emb)
 
 def get_matches(input):
   global index, model, rf