hitz02 commited on
Commit
c99d753
·
1 Parent(s): 9e81777

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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())