Update app.py

app.py

@@ -125,7 +125,7 @@ def fn(    text1,    text2,    num,    slider1,    slider2,    single_checkbox,
     
     
     clinical_dataset = clinical_dataset.map(concatenate_text)
-    embedding = get_embeddings(comments_dataset["text"][0])
+    embedding = get_embeddings(clinical_dataset["text"][0])
     embedding.shape
     
     from transformers import AutoTokenizer, TFAutoModel
@@ -136,7 +136,7 @@ def fn(    text1,    text2,    num,    slider1,    slider2,    single_checkbox,
     
     TensorShape([1, 768])
     
-    embeddings_dataset = comments_dataset.map(
+    embeddings_dataset = clinical_dataset.map(
     lambda x: {"embeddings": get_embeddings(x["text"]).numpy()[0]})
     
     embeddings_dataset.add_faiss_index(column="embeddings")