Update app.py

app.py

@@ -14,8 +14,8 @@ st.title('KoGPT2 Demo')
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 from transformers import AutoTokenizer, AutoModelWithLMHead
-tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPT2MediumProCon")
-model = AutoModelWithLMHead.from_pretrained("gpt2")
+model = AutoModelWithLMHead.from_pretrained("BigSalmon/GPT2MediumProCon")
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
 
 with st.form(key='my_form'):
     prompt = st.text_input(label='Enter sentence')