Update tapas_utils.py

tapas_utils.py

@@ -2,6 +2,15 @@ from transformers import AutoTokenizer, AutoModelForTableQuestionAnswering
 import pandas as pd
 from io import StringIO
 
+from sentence_transformers import SentenceTransformer
+
+retriever = SentenceTransformer("deepset/all-mpnet-base-v2-table")
+
+def embed_table(table):
+    processed_table = "\n".join([table.to_csv(index=False)])
+    return retriever.encode(processed_table)
+
+
 def initialize_tapas():
     tokenizer = AutoTokenizer.from_pretrained("google/tapas-large-finetuned-wtq")
     model = AutoModelForTableQuestionAnswering.from_pretrained("google/tapas-large-finetuned-wtq")