diff --git a/gptc/model.py b/gptc/model.py index 845c3f6..54695f4 100644 --- a/gptc/model.py +++ b/gptc/model.py @@ -95,7 +95,10 @@ class Model: else: explanation[word] = { "weight": weight, - "probabilities": word_probs, + "probabilities": { + name: word_probs[index] + for index, name in enumerate(self.names) + }, "count": 1, } diff --git a/models/compiled.gptc b/models/compiled.gptc index 93502d8..b941678 100644 Binary files a/models/compiled.gptc and b/models/compiled.gptc differ