mlkorra commited on
Commit
378e869
1 Parent(s): b19dd57

update app

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -17,8 +17,7 @@ def load_model(masked_text,model_name):
17
 
18
  result_sentence = nlp(masked_text)
19
 
20
- return result_sentence
21
-
22
 
23
  def main():
24
 
@@ -49,8 +48,10 @@ 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
 
17
 
18
  result_sentence = nlp(masked_text)
19
 
20
+ return result_sentence[0]['sequence']
 
21
 
22
  def main():
23
 
48
  selected_model = models[0]
49
 
50
  if st.button('Fill the Mask!'):
51
+
52
+ with st.spinner("Filling ... "):
53
+ filled_sentence = load_model(masked_text,selected_model)
54
+
55
  st.write(filled_sentence)
56
 
57