Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import tensorflow as tf
 config = tf.compat.v1.ConfigProto(intra_op_parallelism_threads=3, 
                         inter_op_parallelism_threads=2, 
@@ -25,8 +25,8 @@ if st.button('Submit'):
     #get transformers
     from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
 
-    tokenizer = GPT2Tokenizer.from_pretrained("gpt-2")
-    GPT2 = TFGPT2LMHeadModel.from_pretrained("gpt-2", pad_token_id=tokenizer.eos_token_id)
+    tokenizer = AutoTokenizer.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
+    GPT2 = model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
 
     import tensorflow as tf
     tf.random.set_seed(SEED)