jskinner215 commited on
Commit
cc27c6f
·
1 Parent(s): 1233a10

Update tapas_utils.py

Browse files
Files changed (1) hide show
  1. 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")