Spaces:

m-ric
/

attribution_visualizer

Runtime error

m-ric HF staff commited on 3 days ago

Commit

45c11ea

1 Parent(s): d4bacf6

Fillna relevances with 0

Files changed (1) hide show

app.py CHANGED Viewed

@@ -101,7 +101,7 @@ def generate_heatmap(input_tokens, all_relevances, output_tokens):
 def on_generate(input_text, num_tokens):
     input_tokens, relevances, generated_text, generated_length, output_tokens = generate_and_visualize(input_text, num_tokens)
-    heatmap = generate_heatmap(input_tokens, relevances, output_tokens)
     return (
         input_tokens,
         relevances,

 def on_generate(input_text, num_tokens):
     input_tokens, relevances, generated_text, generated_length, output_tokens = generate_and_visualize(input_text, num_tokens)
+    heatmap = generate_heatmap(input_tokens, relevances.fillna(0), output_tokens)
     return (
         input_tokens,
         relevances,