Simon Salmon commited on
Commit
185e7b1
·
1 Parent(s): dfb525f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.text_input(label='Enter sentence')
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: