Spaces:
Build error
Build error
Commit
·
cc27c6f
1
Parent(s):
1233a10
Update tapas_utils.py
Browse files- tapas_utils.py +2 -0
tapas_utils.py
CHANGED
@@ -10,6 +10,8 @@ def embed_table(table):
|
|
10 |
processed_table = "\n".join([table.to_csv(index=False)])
|
11 |
return retriever.encode(processed_table)
|
12 |
|
|
|
|
|
13 |
|
14 |
def initialize_tapas():
|
15 |
tokenizer = AutoTokenizer.from_pretrained("google/tapas-large-finetuned-wtq")
|
|
|
10 |
processed_table = "\n".join([table.to_csv(index=False)])
|
11 |
return retriever.encode(processed_table)
|
12 |
|
13 |
+
def embed_question(question):
|
14 |
+
return retriever.encode(question)
|
15 |
|
16 |
def initialize_tapas():
|
17 |
tokenizer = AutoTokenizer.from_pretrained("google/tapas-large-finetuned-wtq")
|