Meena commited on
Commit
af8985f
·
1 Parent(s): 25b2a82

Update app/tapas.py

Browse files
Files changed (1) hide show
  1. app/tapas.py +1 -2
app/tapas.py CHANGED
@@ -99,12 +99,11 @@ def show_answers(queries, answers, aggregation_predictions_string):
99
  result = "Predicted answer: " + predicted_agg + " > " + answer
100
  return result
101
 
102
- def execute_query(query, table):
103
 
104
  """
105
  Invoke the TAPAS model.
106
  """
107
- queries = [query]
108
  tokenizer, model = load_model_and_tokenizer()
109
  table, inputs = prepare_inputs(table, queries, tokenizer)
110
  predicted_table_cell_coords, predicted_aggregation_operators = generate_predictions(inputs, model, tokenizer)
 
99
  result = "Predicted answer: " + predicted_agg + " > " + answer
100
  return result
101
 
102
+ def execute_query(queries, table):
103
 
104
  """
105
  Invoke the TAPAS model.
106
  """
 
107
  tokenizer, model = load_model_and_tokenizer()
108
  table, inputs = prepare_inputs(table, queries, tokenizer)
109
  predicted_table_cell_coords, predicted_aggregation_operators = generate_predictions(inputs, model, tokenizer)