Upload QApipeline

QBModelConfig.py

@@ -6,4 +6,4 @@ class QBModelConfig(PretrainedConfig):
 
     def __init__(self, **kwargs):
         super().__init__( **kwargs)
-        self.torch_dtype = torch.float16
+        self.torch_dtype = torch.float32

QBModelWrapper.py

@@ -3,10 +3,20 @@ from transformers import PreTrainedModel
 from transformers import PretrainedConfig
 from QBModelConfig import QBModelConfig
 from qbmodel import QuizBowlModel
+from huggingface_hub import hf_hub_download
+
+REPO_ID = "Backedman/TriviaAnsweringMachineREAL"
+FILENAME = "models/Mythology_tfidf.pkl"
 
 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='')
+
+
 
     def __init__(self, config):
         super().__init__(config)

README.md

@@ -1,7 +1,7 @@
 ---
-license: mit
 language:
 - en
+license: mit
 pipeline_tag: question-answering
 ---
 The evaluation of this project is to answer trivia questions.  You do