Update app.py
Browse files
app.py
CHANGED
@@ -17,10 +17,10 @@ artist_name = st.text_input("Model", "roberta-base")
|
|
17 |
tokenizer = AutoTokenizer.from_pretrained("roberta-base")
|
18 |
model = AutoModelForMaskedLM.from_pretrained(artist_name)
|
19 |
|
20 |
-
|
21 |
|
22 |
with st.form(key='my_form'):
|
23 |
-
prompt = st.text_area(label='Enter Text. Put <mask> where you want the model to fill in the blank. You can use more than one at a time.')
|
24 |
submit_button = st.form_submit_button(label='Submit')
|
25 |
|
26 |
if submit_button:
|
|
|
17 |
tokenizer = AutoTokenizer.from_pretrained("roberta-base")
|
18 |
model = AutoModelForMaskedLM.from_pretrained(artist_name)
|
19 |
|
20 |
+
first = "Boston is a <mask> place to live."
|
21 |
|
22 |
with st.form(key='my_form'):
|
23 |
+
prompt = st.text_area(label='Enter Text. Put <mask> where you want the model to fill in the blank. You can use more than one at a time.', value=first)
|
24 |
submit_button = st.form_submit_button(label='Submit')
|
25 |
|
26 |
if submit_button:
|