Create app.py

app.py

@@ -0,0 +1,44 @@
+import streamlit as st
+
+#for reproducability
+SEED = 12
+
+#maximum number of words in output text
+# MAX_LEN = 30
+
+title = st.text_input('Enter the seed words', ' ')
+input_sequence = title
+
+number = st.number_input('Insert how many words', 1)
+MAX_LEN = number
+
+if st.button('Submit'):
+
+    #get transformers
+    from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
+
+    tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
+    GPT2 = TFGPT2LMHeadModel.from_pretrained("gpt2-medium", pad_token_id=tokenizer.eos_token_id)
+
+    import tensorflow as tf
+    tf.random.set_seed(SEED)
+    input_ids = tokenizer.encode(input_sequence, return_tensors='tf')
+
+    #sample only from 80% most likely words
+    sample_output = GPT2.generate(
+                                input_ids, 
+                                do_sample = True, 
+                                max_length = MAX_LEN, 
+                                top_p = 0.8, 
+                                top_k = 0
+    )
+
+    st.write(tokenizer.decode(sample_output[0], skip_special_tokens = True))
+
+else:
+    st.write(' ')
+
+
+
+# print("Output:\n" + 100 * '-')
+# print(tokenizer.decode(sample_output[0], skip_special_tokens = True), '...')