mlkorra commited on
Commit
1664596
1 Parent(s): d36be85

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,7 +17,7 @@ def load_model(masked_text,model_name):
17
 
18
  result_sentence = nlp(masked_text)
19
 
20
- return result_sentence
21
 
22
  def main():
23
 
@@ -42,7 +42,7 @@ def main():
42
  st.sidebar.title("Hindi MLM")
43
  masked_text = st.sidebar.selectbox('Select any of the following text',
44
  texts)
45
-
46
  st.write('You selected:', masked_text)
47
 
48
  selected_model = models[0]
17
 
18
  result_sentence = nlp(masked_text)
19
 
20
+ return result_sentence[0]['sequence']
21
 
22
  def main():
23
 
42
  st.sidebar.title("Hindi MLM")
43
  masked_text = st.sidebar.selectbox('Select any of the following text',
44
  texts)
45
+
46
  st.write('You selected:', masked_text)
47
 
48
  selected_model = models[0]