Update handler.py

handler.py

@@ -38,8 +38,11 @@ class EndpointHandler:
                 topic_info = self.topic_model.get_topic(topic)
                 topic_words = [word for word, _ in topic_info] if topic_info else []
 
-                # Get custom label for the topic
-                custom_label = self.topic_model.custom_labels_[topic + 1] if hasattr(self.topic_model, "custom_labels_") else f"Topic {topic}"
+                # Get custom label for the topic (with fallback if custom_labels_ is not available)
+                if hasattr(self.topic_model, "custom_labels_") and self.topic_model.custom_labels_ is not None:
+                    custom_label = self.topic_model.custom_labels_[topic + 1]
+                else:
+                    custom_label = f"Topic {topic}"  # Fallback label
 
                 results.append({
                     "topic": int(topic),
@@ -76,3 +79,5 @@ class EndpointHandler:
         except Exception as e:
             return [{"error": str(e)}]  # Returning error as a list with a dictionary
 
+# Create an instance of the handler
+handler = EndpointHandler()