sohamsh commited on
Commit
2864b0a
1 Parent(s): ca871e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -19,6 +19,7 @@ def get_model():
19
  GPT2 = TFGPT2LMHeadModel.from_pretrained("gpt2-medium", pad_token_id=tokenizer.eos_token_id)
20
  return GPT2, tokenizer
21
 
 
22
  c = 5
23
  with st.form(key='my_form'):
24
  prompt = st.text_input('Enter sentence:', '')
 
19
  GPT2 = TFGPT2LMHeadModel.from_pretrained("gpt2-medium", pad_token_id=tokenizer.eos_token_id)
20
  return GPT2, tokenizer
21
 
22
+ GPT2, tokenizer = get_model()
23
  c = 5
24
  with st.form(key='my_form'):
25
  prompt = st.text_input('Enter sentence:', '')