Update app.py

app.py

@@ -2,14 +2,16 @@ import streamlit as st
 from streamlit.elements.altair import generate_chart
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from transformers import pipeline
-st.title("Rap Lyrics Generator")
-artist = st.text_input("Enter the artist", "Eminem")
-song_name = st.text_input("Enter the desired song name", "Gas is going")
-
+st.title("Loading model")
 model_ckpt = "flax-community/gpt2-rap-lyric-generator"
 tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
 model = AutoModelForCausalLM.from_pretrained(model_ckpt,from_flax=True)
 text_generation = pipeline("text-generation", model=model, tokenizer=tokenizer)
+st.title("Rap lyrics generator")
+artist = st.text_input("Enter the artist", "Eminem")
+song_name = st.text_input("Enter the desired song name", "Gas is going")
+
+
 
 if st.button("Generate lyrics"):
     st.title(f"{artist}: {song_name}")