Meena commited on
Commit
8947029
1 Parent(s): 4a8f9f2

Update app/tapas.py

Browse files
Files changed (1) hide show
  1. app/tapas.py +2 -2
app/tapas.py CHANGED
@@ -90,12 +90,12 @@ def show_answers(queries, answers, aggregation_predictions_string):
90
  results.append(result)
91
  return results
92
 
93
- def execute_query(queries, table):
94
 
95
  """
96
  Invoke the TAPAS model.
97
  """
98
- print(queries)
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)
 
90
  results.append(result)
91
  return results
92
 
93
+ def execute_query(query, table):
94
 
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)