Backedman
/

TriviaAnsweringMachineREAL

Question Answering

Model card Files Files and versions Community

Backedman commited on May 7

Commit

38fdb90

•

1 Parent(s): c7802ab

Update qbmodel.py

Files changed (1) hide show

qbmodel.py +1 -1

qbmodel.py CHANGED Viewed

@@ -72,7 +72,7 @@ class QuizBowlModel:
                 if self.tfidf_models[category] is None:
                     self.tfidf_models[category] = NLPModel().load(f"models/{self.categories[category]}_tfidf.pkl")
-            self.tfidf_models[-1] = NLPModel().load(f"models/{'ALL'}_tfidf.pkl")
         else:
             for category in range(len(self.categories)):
                 if self.tfidf_models[category] is None:

                 if self.tfidf_models[category] is None:
                     self.tfidf_models[category] = NLPModel().load(f"models/{self.categories[category]}_tfidf.pkl")
+            #self.tfidf_models[-1] = NLPModel().load(f"models/{'ALL'}_tfidf.pkl")
         else:
             for category in range(len(self.categories)):
                 if self.tfidf_models[category] is None: