Update app/tapas.py

app/tapas.py

@@ -4,8 +4,6 @@ import re
 
 p = re.compile('\d+(\.\d+)?')
 
-queries = ["Rating Of Breaking Bad?","Rating Of The God Father?"]
-
 def load_model_and_tokenizer():
     """
     Load
@@ -96,7 +94,7 @@ def execute_query(query, table):
 	"""
 	  Invoke the TAPAS model.
 	"""
-#	queries = query.split(',')
+    queries = [query]
 	tokenizer, model = load_model_and_tokenizer()
 	table, inputs = prepare_inputs(table, queries, tokenizer)
 	predicted_table_cell_coords, predicted_aggregation_operators = generate_predictions(inputs, model, tokenizer)