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)
|