hitz02 commited on
Commit
227b705
1 Parent(s): c99d753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -15,13 +15,6 @@ from transformers import TapasTokenizer, TapasForQuestionAnswering
15
 
16
  tf.get_logger().setLevel('ERROR')
17
 
18
- #try:
19
- #print(sys.executable)
20
- # subprocess.check_call(["/home/user/.local/lib/python3.8", "-m", "pip", "install", 'torch-scatter','-f', 'https://data.pyg.org/whl/torch-1.10.0+cpu.html'])
21
- #except Exception as e:
22
- # print('Error..', str(e))
23
-
24
-
25
  model_name = 'google/tapas-base-finetuned-wtq'
26
  #model_name = "table-question-answering"
27
  #model = pipeline(model_name)
@@ -59,8 +52,8 @@ def styling_specific_cell(x,tags,colors):
59
 
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())
 
15
 
16
  tf.get_logger().setLevel('ERROR')
17
 
 
 
 
 
 
 
 
18
  model_name = 'google/tapas-base-finetuned-wtq'
19
  #model_name = "table-question-answering"
20
  #model = pipeline(model_name)
 
52
 
53
  if st.button('Predict Answers'):
54
  with st.spinner('It will take approx a minute'):
55
+ table = data.astype(str)
56
+ inputs = tokenizer(table=table , queries=input_queries, padding='max_length',truncation=True, return_tensors="pt")
57
  outputs = model(**inputs)
58
  #outputs = model(table = data, query = queries)
59
  predicted_answer_coordinates, predicted_aggregation_indices = tokenizer.convert_logits_to_predictions( inputs, outputs.logits.detach(), outputs.logits_aggregation.detach())