Update QBModelWrapper.py
Browse files- QBModelWrapper.py +3 -3
QBModelWrapper.py
CHANGED
@@ -12,9 +12,9 @@ class QBModelWrapper(PreTrainedModel):
|
|
12 |
config_class= QBModelConfig
|
13 |
config = QBModelConfig
|
14 |
|
15 |
-
hf_hub_download(repo_id=REPO_ID, filename='tfidf_model.py', local_dir='')
|
16 |
-
hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='')
|
17 |
-
hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='')
|
18 |
|
19 |
|
20 |
|
|
|
12 |
config_class= QBModelConfig
|
13 |
config = QBModelConfig
|
14 |
|
15 |
+
hf_hub_download(repo_id=REPO_ID, filename='tfidf_model.py', local_dir='.')
|
16 |
+
hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='.')
|
17 |
+
hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='.')
|
18 |
|
19 |
|
20 |
|