SCANSKY commited on
Commit
b8f2fb6
·
verified ·
1 Parent(s): 9179af7

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -64,7 +64,7 @@ class EndpointHandler:
64
  "probability": float(prob),
65
  "top_words": topic_words[:5], # Top 5 words
66
  "customLabel": custom_label, # Add custom label
67
- "contribution": float(contribution) # Add contribution from approximate distribution
68
  })
69
 
70
  return results
 
64
  "probability": float(prob),
65
  "top_words": topic_words[:5], # Top 5 words
66
  "customLabel": custom_label, # Add custom label
67
+ "contribution": ranked_topics
68
  })
69
 
70
  return results