Update app.py

app.py

@@ -23,7 +23,7 @@ def plot_word_scores(top_token_ids, outputs, tokenizer, boolq=False, width=600):
     for step, candidates in enumerate(top_token_ids):  
         fig.append_trace(
             go.Bar(
-                y=tokenizer.convert_ids_to_tokens(candidates), 
+                y=[w[1:] for w in tokenizer.convert_ids_to_tokens(candidates)], 
                 x=outputs['scores'][step][0][candidates], 
                 orientation='h'
             ),
@@ -51,7 +51,7 @@ with col2:
     top_p = st.slider('Top p:', min_value=0.5, max_value=1.0, value=0.99)
     # max_tokens = st.number_input('Max output length:', min_value=1, max_value=64, format='%i')
     max_tokens = st.slider('Max output length: ', min_value=1, max_value=64)
-    threshold = st.number_input('Min token score:: ', value=-100.0)
+    # threshold = st.number_input('Min token score:: ', value=-10.0)
     
     
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)