Merge branch 'main' of https://huggingface.co/spaces/flax-community/roberta-hindi into main

app.py

@@ -11,19 +11,19 @@ def load_model(masked_text,model_name):
 
     model = AutoModelForMaskedLM.from_pretrained(model_name, from_flax=True)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    tokenizer.save_pretrained('exported_pytorch_model')
-    model.save_pretrained('exported_pytorch_model')
-    nlp = pipeline('fill-mask', model="exported_pytorch_model")
+    # tokenizer.save_pretrained('exported_pytorch_model')
+    # model.save_pretrained('exported_pytorch_model')
+    nlp = pipeline('fill-mask', model=model, tokenizer=tokenizer)
 
     result_sentence = nlp(masked_text)
 
-    return result_sentence[0]['sequence']
+    return result_sentence
 
 def main():
 
-    st.title("RoBERTa-Hindi")
+    st.title("RoBERTa Hindi")
     st.markdown(
-    "This demo uses pretrained RoBERTa variants for Mask Language Modelling (MLM)"
+    "This demo uses pretrained RoBERTa variants for Mask Language Modeling (MLM)"
     )
 
     models = st.multiselect(
@@ -48,11 +48,9 @@ def main():
     selected_model = models[0]
 
     if st.button('Fill the Mask!'):
-
-        with st.spinner("Filling ... "):  
+         with st.spinner("Filling the Mask..."):
             filled_sentence = load_model(masked_text,selected_model)
-
-        st.write(filled_sentence)
+            st.write(filled_sentence)
 
     
 if __name__ == "__main__":