Shankhdhar commited on
Commit
1d7eda9
·
1 Parent(s): 4744267

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelWithLMHead
3
  from transformers import pipeline
4
  st.title("Rap Lyrics Generator")
5
- st.image('./parental-advisory_wide-1aa5c9539fe4e7f1ad184c7d75ab33f1c07c0038.jpeg')
6
  model_ckpt = "flax-community/gpt2-rap-lyric-generator"
7
  tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
8
  model = AutoModelWithLMHead.from_pretrained(model_ckpt,from_flax=True)
 
2
  from transformers import AutoTokenizer, AutoModelWithLMHead
3
  from transformers import pipeline
4
  st.title("Rap Lyrics Generator")
5
+ st.image('./parental.png')
6
  model_ckpt = "flax-community/gpt2-rap-lyric-generator"
7
  tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
8
  model = AutoModelWithLMHead.from_pretrained(model_ckpt,from_flax=True)