Updated inference.py

apps/inference.py

@@ -6,6 +6,7 @@ from transformers import pipeline
 import os
 import json
 
+
 models = []
 predicted_tokens = []
 predicted_sentence = []
@@ -53,9 +54,18 @@ def app():
         ],
         ["flax-community/roberta-hindi"],
     )
-
     if st.button("Fill the Mask!"):
         with st.spinner("Filling the Mask..."):
-            st.write(selected_models)
-            # filled_sentence = load_model(masked_text, selected_model)
+
+            for i in range(len(selected_models)):
+                filled_sentence = load_model(masked_text, selected_models[i])
+                models.append(selected_models[i])
+                predicted_tokens.append(filled_sentence["token_str"])
+                predicted_sentence.append(filled_sentence["sequence"])
+
+            results_df = pd.DataFrame()
+            results_df["Model Name"] = models
+            results_df["Predicted Word"] = predicted_tokens
+            results_df["Sentence"] = predicted_sentence
+            st.dataframe(results_df)