Update app.py

app.py

@@ -20,7 +20,7 @@ if st.button('Submit'):
     from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
 
     tokenizer = AutoTokenizer.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
-    GPT2 = model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
+    model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
     tf.random.set_seed(SEED)
     
     input_ids = tokenizer.encode(input_sequence, return_tensors='tf')
@@ -30,7 +30,7 @@ if st.button('Submit'):
     outputs = model(**inputs)
 
     print("Output:\n" + 100 * '-')
-    print(tokenizer.decode(outputs[0], skip_special_tokens = True))
+    print(outputs)
 else:
     st.write(' ')