Backedman
/

TriviaAnsweringMachineREAL

Question Answering

Model card Files Files and versions

Backedman commited on May 7, 2024

Commit

c7802ab

·

verified ·

1 Parent(s): 0a03546

Update qbmodel.py

Files changed (1) hide show

qbmodel.py +1 -1

qbmodel.py CHANGED Viewed

@@ -26,7 +26,7 @@ class QuizBowlModel:
         as it will increase latency severely.
         """
-        self.categories = ['Geography', 'Religion', 'Philosophy', 'Trash','Mythology', 'Literature','Science', 'Social Science', 'History', 'Current Events', 'Fine Arts', 'ALL']
         self.tfidf_models = [None for _ in range(len(self.categories))]
         self.qc_model = qc.TextClassificationModel.load_model("models/categorizer")

         as it will increase latency severely.
         """
+        self.categories = ['Geography', 'Religion', 'Philosophy', 'Trash','Mythology', 'Literature','Science', 'Social Science', 'History', 'Current Events', 'Fine Arts']
         self.tfidf_models = [None for _ in range(len(self.categories))]
         self.qc_model = qc.TextClassificationModel.load_model("models/categorizer")