Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ def styling_specific_cell(x,tags,colors):
|
|
60 |
if st.button('Predict Answers'):
|
61 |
with st.spinner('It will take approx a minute'):
|
62 |
data = data.astype(str)
|
63 |
-
inputs = tokenizer(table=data, queries=input_queries, padding='max_length', return_tensors="pt")
|
64 |
outputs = model(**inputs)
|
65 |
#outputs = model(table = data, query = queries)
|
66 |
predicted_answer_coordinates, predicted_aggregation_indices = tokenizer.convert_logits_to_predictions( inputs, outputs.logits.detach(), outputs.logits_aggregation.detach())
|
|
|
60 |
if st.button('Predict Answers'):
|
61 |
with st.spinner('It will take approx a minute'):
|
62 |
data = data.astype(str)
|
63 |
+
inputs = tokenizer(table=data, queries=input_queries, padding='max_length',truncation=True, return_tensors="pt")
|
64 |
outputs = model(**inputs)
|
65 |
#outputs = model(table = data, query = queries)
|
66 |
predicted_answer_coordinates, predicted_aggregation_indices = tokenizer.convert_logits_to_predictions( inputs, outputs.logits.detach(), outputs.logits_aggregation.detach())
|