foxxy-hm
/

e2eqa-wiki

Question Answering

Model card Files Files and versions

foxxy-hm commited on Jul 14, 2023

Commit

3edaf17

·

1 Parent(s): 705e36e

Update handler.py

Files changed (1) hide show

handler.py +3 -2

handler.py CHANGED Viewed

@@ -3,10 +3,11 @@ from features.text_utils import *
 import regex as re
 from models.bm25_utils import BM25Gensim
 from models.qa_model import *
-# from tqdm.auto import tqdm
-# tqdm.pandas()
 from datasets import load_dataset
 from transformers import pipeline
 class InferencePipeline():
     def __init__(self, path=""):

 import regex as re
 from models.bm25_utils import BM25Gensim
 from models.qa_model import *
+from tqdm.auto import tqdm
+tqdm.pandas()
 from datasets import load_dataset
 from transformers import pipeline
+from typing import Dict, List, Any
 class InferencePipeline():
     def __init__(self, path=""):