mlkorra commited on
Commit
decd5de
1 Parent(s): 1664596

update app

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -45,12 +45,18 @@ def main():
45
 
46
  st.write('You selected:', masked_text)
47
 
48
- selected_model = models[0]
49
 
50
- if st.button('Fill the Mask!'):
51
- with st.spinner("Filling the Mask..."):
52
- filled_sentence = load_model(masked_text,selected_model)
53
- st.write(filled_sentence)
 
 
 
 
 
 
54
 
55
 
56
  if __name__ == "__main__":
45
 
46
  st.write('You selected:', masked_text)
47
 
48
+ results_df = pd.DataFrame(columns = ['Model Name','Masked Text','Filled Masked Text'])
49
 
50
+ for selected_model in models:
51
+
52
+ if st.button('Fill the Mask!'):
53
+ with st.spinner("Filling the Mask..."):
54
+ filled_sentence = load_model(masked_text,selected_model)
55
+ results_df['Model Name'] = selected_model
56
+ results_df['Masked Text'] = masked_text
57
+ results_df['Filled Masked Text'] = filled_sentence
58
+
59
+ st.table(results_df)
60
 
61
 
62
  if __name__ == "__main__":