Cropinky commited on
Commit
2620132
·
1 Parent(s): fe24050

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
8
  model = AutoModelForCausalLM.from_pretrained(model_ckpt,from_flax=True)
9
  text_generation = pipeline("text-generation", model=model, tokenizer=tokenizer)
10
  title.title("Rap lyrics generator")
11
- artist = st.text_input("Enter the artist", "Raekwon")
12
- song_name = st.text_input("Enter the desired song name", "Running Them Streets")
13
 
14
 
15
 
 
8
  model = AutoModelForCausalLM.from_pretrained(model_ckpt,from_flax=True)
9
  text_generation = pipeline("text-generation", model=model, tokenizer=tokenizer)
10
  title.title("Rap lyrics generator")
11
+ artist = st.text_input("Enter the artist", "Wu-Tang Clan")
12
+ song_name = st.text_input("Enter the desired song name", "New York Gangstas")
13
 
14
 
15