foxxy-hm commited on
Commit
3edaf17
·
1 Parent(s): 705e36e

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -3,10 +3,11 @@ from features.text_utils import *
3
  import regex as re
4
  from models.bm25_utils import BM25Gensim
5
  from models.qa_model import *
6
- # from tqdm.auto import tqdm
7
- # tqdm.pandas()
8
  from datasets import load_dataset
9
  from transformers import pipeline
 
10
 
11
  class InferencePipeline():
12
  def __init__(self, path=""):
 
3
  import regex as re
4
  from models.bm25_utils import BM25Gensim
5
  from models.qa_model import *
6
+ from tqdm.auto import tqdm
7
+ tqdm.pandas()
8
  from datasets import load_dataset
9
  from transformers import pipeline
10
+ from typing import Dict, List, Any
11
 
12
  class InferencePipeline():
13
  def __init__(self, path=""):