Meena commited on
Commit
e79a33c
·
1 Parent(s): 32fe53d

Update app/tapas.py

Browse files
Files changed (1) hide show
  1. app/tapas.py +2 -1
app/tapas.py CHANGED
@@ -4,6 +4,7 @@ import re
4
 
5
  p = re.compile('\d+(\.\d+)?')
6
 
 
7
 
8
  def load_model_and_tokenizer():
9
  """
@@ -95,7 +96,7 @@ def execute_query(query, table):
95
  """
96
  Invoke the TAPAS model.
97
  """
98
- queries = query.split(',')
99
  tokenizer, model = load_model_and_tokenizer()
100
  table, inputs = prepare_inputs(table, queries, tokenizer)
101
  predicted_table_cell_coords, predicted_aggregation_operators = generate_predictions(inputs, model, tokenizer)
 
4
 
5
  p = re.compile('\d+(\.\d+)?')
6
 
7
+ queries = ["Rating Of Breaking Bad?","Rating Of The God Father?"]
8
 
9
  def load_model_and_tokenizer():
10
  """
 
96
  """
97
  Invoke the TAPAS model.
98
  """
99
+ # queries = query.split(',')
100
  tokenizer, model = load_model_and_tokenizer()
101
  table, inputs = prepare_inputs(table, queries, tokenizer)
102
  predicted_table_cell_coords, predicted_aggregation_operators = generate_predictions(inputs, model, tokenizer)