Update qbmodel.py

qbmodel.py

@@ -67,6 +67,12 @@ class QuizBowlModel:
         # Create respective model if not exist
         if not clear:
             for category in range(len(self.categories)):
+
+                REPO_ID = 'Backedman/TriviaAnsweringMachineREAL'
+                FILENAME = f"models/{self.categories[category]}_tfidf.pkl"
+
+                hf_hub_download(repo_id=REPO_ID, filename=FILENAME, local_dir='.')
+                
                 if self.tfidf_models[category] is None:
                     self.tfidf_models[category] = NLPModel().load(f"models/{self.categories[category]}_tfidf.pkl")