updated inference.py

apps/inference.py

@@ -6,6 +6,10 @@ from transformers import pipeline
 import os
 import json
 
+models = []
+predicted_tokens = []
+predicted_sentence = []
+
 
 @st.cache(show_spinner=False, persist=True)
 def load_model(masked_text, model_name):
@@ -38,7 +42,7 @@ def app():
 
     st.write("You selected:", masked_text)
 
-    models = st.multiselect(
+    selected_models = st.multiselect(
         "Choose models",
         [
             "flax-community/roberta-hindi",
@@ -50,9 +54,8 @@ def app():
         ["flax-community/roberta-hindi"],
     )
 
-    selected_model = models[0]
-
     if st.button("Fill the Mask!"):
         with st.spinner("Filling the Mask..."):
-            filled_sentence = load_model(masked_text, selected_model)
-            st.write(filled_sentence)
+            st.write(selected_models)
+            # filled_sentence = load_model(masked_text, selected_model)
+