hassiahk commited on
Commit
90dc487
1 Parent(s): a8ac4ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -22,9 +22,9 @@ def load_model(masked_text,model_name):
22
 
23
  def main():
24
 
25
- st.title("RoBERTa-Hindi")
26
  st.markdown(
27
- "This demo uses pretrained RoBERTa variants for Mask Language Modelling (MLM)"
28
  )
29
 
30
  models = st.multiselect(
@@ -49,9 +49,9 @@ def main():
49
  selected_model = models[0]
50
 
51
  if st.button('Fill the Mask!'):
52
-
53
- filled_sentence = load_model(masked_text,selected_model)
54
- st.write(filled_sentence)
55
 
56
 
57
  if __name__ == "__main__":
22
 
23
  def main():
24
 
25
+ st.title("RoBERTa Hindi")
26
  st.markdown(
27
+ "This demo uses pretrained RoBERTa variants for Mask Language Modeling (MLM)"
28
  )
29
 
30
  models = st.multiselect(
49
  selected_model = models[0]
50
 
51
  if st.button('Fill the Mask!'):
52
+ with st.spinner("Filling the Mask..."):
53
+ filled_sentence = load_model(masked_text,selected_model)
54
+ st.write(filled_sentence)
55
 
56
 
57
  if __name__ == "__main__":