Backedman
/

TriviaAnsweringMachineREAL

Question Answering

Model card Files Files and versions

Backedman commited on May 7, 2024

Commit

7c35b20

·

verified ·

1 Parent(s): 20dcf24

Update QBModelWrapper.py

Files changed (1) hide show

QBModelWrapper.py +3 -3

QBModelWrapper.py CHANGED Viewed

@@ -12,9 +12,9 @@ class QBModelWrapper(PreTrainedModel):
     config_class= QBModelConfig
     config = QBModelConfig
-    hf_hub_download(repo_id=REPO_ID, filename='tfidf_model.py', local_dir='')
-    hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='')
-    hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='')

     config_class= QBModelConfig
     config = QBModelConfig
+    hf_hub_download(repo_id=REPO_ID, filename='tfidf_model.py', local_dir='.')
+    hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='.')
+    hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='.')