Create app.py

app.py

@@ -0,0 +1,10 @@
+import streamlit as st
+from transformers import AutoTokenizer, AutoModelWithLMHead
+from transformers import pipeline
+model_ckpt = "flax-community/gpt2-rap-lyric-generator"
+tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
+model = AutoModelWithLMHead.from_pretrained(model_ckpt,from_flax=True)
+text_generation = pipeline("text-generation", model=model, tokenizer=tokenizer)
+prefix_text = st.text_input("Enter the prefix text", default_value_goes_here)
+generated_text= text_generation(prefix_text, max_length=500, do_sample=True)[0]
+st.write(generated_text['generated_text'])