Update app.py

app.py

@@ -32,12 +32,11 @@ sent += ' <mask>'
 pipe = pipeline("fill-mask", tokenizer = tokenizer, model = model)
 dict_next_words = next_word(sent, pipe)
 df = pd.DataFrame.from_dict(dict_next_words)
-df.reset_index(drop=True, inplace=True)
 #design the graph theme
 sns.set_theme(style="whitegrid")
 sns.set_color_codes("pastel")
 fig, ax = plt.subplots(figsize=(6, 15))
-sns.barplot(data=df, orient = 'h')
+sns.barplot(x='Score', y='Word', hue='level_0', data=df, orient = 'h')
 st.pyplot(fig)
 
 #st.table(df)