Spaces:
Build error
Build error
Update app/tapas.py
Browse files- 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(
|
94 |
|
95 |
"""
|
96 |
Invoke the TAPAS model.
|
97 |
"""
|
98 |
-
|
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)
|