RohitMidha23 commited on
Commit
de5d62c
1 Parent(s): 6b9824d

module error fix

Browse files
Files changed (1) hide show
  1. src/inference.py +1 -1
src/inference.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import TapasTokenizer, TapasForQuestionAnswering
2
  import pandas as pd
3
  from typing import List, Dict
4
- from constants import id2aggregation
5
 
6
  def infer(query: str, file_name: str, model_name: str="google/tapas-base-finetuned-wtq") -> Dict[str, str]:
7
  # Load the file
 
1
  from transformers import TapasTokenizer, TapasForQuestionAnswering
2
  import pandas as pd
3
  from typing import List, Dict
4
+ from src.constants import id2aggregation
5
 
6
  def infer(query: str, file_name: str, model_name: str="google/tapas-base-finetuned-wtq") -> Dict[str, str]:
7
  # Load the file