Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -22,9 +22,9 @@ def load_model(masked_text,model_name):
|
|
22 |
|
23 |
def main():
|
24 |
|
25 |
-
st.title("RoBERTa
|
26 |
st.markdown(
|
27 |
-
"This demo uses pretrained RoBERTa variants for Mask Language
|
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 |
-
|
54 |
-
|
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__":
|