Spaces:
Runtime error
Runtime error
Simon Salmon
commited on
Commit
·
185e7b1
1
Parent(s):
dfb525f
Update app.py
Browse files
app.py
CHANGED
@@ -14,11 +14,11 @@ st.title('KoGPT2 Demo')
|
|
14 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
15 |
|
16 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
17 |
-
model = AutoModelWithLMHead.from_pretrained("BigSalmon/GPT2MediumProCon")
|
18 |
tokenizer = AutoTokenizer.from_pretrained("gpt2")
|
|
|
19 |
|
20 |
with st.form(key='my_form'):
|
21 |
-
prompt = st.
|
22 |
submit_button = st.form_submit_button(label='Submit')
|
23 |
|
24 |
if submit_button:
|
|
|
14 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
15 |
|
16 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
|
|
17 |
tokenizer = AutoTokenizer.from_pretrained("gpt2")
|
18 |
+
model = AutoModelWithLMHead.from_pretrained("BigSalmon/Parentheses")
|
19 |
|
20 |
with st.form(key='my_form'):
|
21 |
+
prompt = st.text_area(label='Enter sentence')
|
22 |
submit_button = st.form_submit_button(label='Submit')
|
23 |
|
24 |
if submit_button:
|