Spaces:
Runtime error
Runtime error
mlkorra
commited on
Commit
·
378e869
1
Parent(s):
b19dd57
update app
Browse files
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 |
-
|
|
|
|
|
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 |
|