Update app.py

app.py

@@ -222,8 +222,15 @@ class NewsGroupsDataLoader:
         all_labels = list(train_data['label']) + list(test_data['label'])
         
         # Get category names from dataset features
-        self.categories = train_data.features['label'].names
-
+        self.categories = [
+            'alt.atheism', 'comp.graphics', 'comp.os.ms-windows.misc',
+            'comp.sys.ibm.pc.hardware', 'comp.sys.mac.hardware', 'comp.windows.x',
+            'misc.forsale', 'rec.autos', 'rec.motorcycles', 'rec.sport.baseball',
+            'rec.sport.hockey', 'sci.crypt', 'sci.electronics', 'sci.med',
+            'sci.space', 'soc.religion.christian', 'talk.politics.guns',
+            'talk.politics.mideast', 'talk.politics.misc', 'talk.religion.misc'
+        ]
+        
         logger.info(f"Total documents: {len(all_texts)}")
         logger.info(f"Number of categories: {len(self.categories)}")
         logger.info(f"Categories: {self.categories}")