update app

app.py

@@ -45,12 +45,18 @@ def main():
     
     st.write('You selected:', masked_text)
     
-    selected_model = models[0]
+    results_df = pd.DataFrame(columns = ['Model Name','Masked Text','Filled Masked Text'])
 
-    if st.button('Fill the Mask!'):
-         with st.spinner("Filling the Mask..."):
-            filled_sentence = load_model(masked_text,selected_model)
-            st.write(filled_sentence)
+    for selected_model in models:
+
+        if st.button('Fill the Mask!'):
+            with st.spinner("Filling the Mask..."):
+                filled_sentence = load_model(masked_text,selected_model)
+                results_df['Model Name'] = selected_model
+                results_df['Masked Text'] = masked_text
+                results_df['Filled Masked Text'] = filled_sentence
+
+    st.table(results_df)
 
     
 if __name__ == "__main__":