Backedman commited on
Commit
7c35b20
1 Parent(s): 20dcf24

Update QBModelWrapper.py

Browse files
Files changed (1) hide show
  1. 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