Upload model

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)

config.json

@@ -7,6 +7,6 @@
     "AutoModelForQuestionAnswering": "QBModelWrapper.QBModelWrapper"
   },
   "model_type": "TFIDF-QA",
-  "torch_dtype": "float16",
+  "torch_dtype": "float32",
   "transformers_version": "4.40.1"
 }