Update app.py

app.py

@@ -21,16 +21,16 @@ if st.button('Submit'):
 
     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)
+    
     input_ids = tokenizer.encode(input_sequence, return_tensors='tf')
 
 # generate text until the output length (which includes the context length) reaches 50
-    greedy_output = GPT2.generate(input_ids, max_length = MAX_LEN)
+    #greedy_output = GPT2.generate(input_ids, max_length = MAX_LEN)
+    outputs = model(**inputs)
 
     print("Output:\n" + 100 * '-')
-    print(tokenizer.decode(greedy_output[0], skip_special_tokens = True))
+    print(tokenizer.decode(outputs[0], skip_special_tokens = True))
 else:
     st.write(' ')