Update app.py

app.py

@@ -25,8 +25,8 @@ if st.button('Submit'):
     #get transformers
     from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
 
-    tokenizer = GPT2Tokenizer.from_pretrained("gpt2-large")
-    GPT2 = TFGPT2LMHeadModel.from_pretrained("gpt2-large", pad_token_id=tokenizer.eos_token_id)
+    tokenizer = GPT2Tokenizer.from_pretrained("ml6team/gpt-2-small-conditional-quote-generator")
+    GPT2 = TFGPT2LMHeadModel.from_pretrained("ml6team/gpt-2-small-conditional-quote-generator", pad_token_id=tokenizer.eos_token_id)
 
     import tensorflow as tf
     tf.random.set_seed(SEED)