Update handler.py
Browse files- 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 |
-
|
7 |
-
|
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=""):
|