Update handler.py

handler.py

@@ -33,17 +33,15 @@ class EndpointHandler:
             # Perform topic inference
             topics, probabilities = self.topic_model.transform(docs)
 
-            # Get topic information
-            topics_info = self.topic_model.get_topics()
-
             # Prepare the results
             results = []
             for topic, prob in zip(topics, probabilities):
-                topic_words = topics_info.get(topic, [])
+                topic_info = self.topic_model.get_topic(topic)
+                topic_words = [word for word, _ in topic_info] if topic_info else []
                 results.append({
                     "topic": int(topic),
                     "probability": float(prob),
-                    "top_words": [word for word, _ in topic_words[:5]]  # Top 5 words
+                    "top_words": topic_words[:5]  # Top 5 words
                 })
 
             return results
@@ -52,11 +50,9 @@ class EndpointHandler:
 
     def postprocess(self, results):
         """
-        Postprocess the inference results into a JSON response.
+        Postprocess the inference results into a JSON-serializable list.
         """
-        return {
-            "results": results
-        }
+        return results  # Directly returning the list of results
 
     def __call__(self, data):
         """
@@ -74,6 +70,4 @@ class EndpointHandler:
 
             return response
         except Exception as e:
-            return {
-                "error": str(e)
-            }
+            return [{"error": str(e)}]  # Returning error as a list with a dictionary