Spaces:
Runtime error
Runtime error
changes
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@ import streamlit as st
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
| 3 |
from transformers import pipeline
|
| 4 |
st.title("Rap Lyrics Generator")
|
|
|
|
| 5 |
model_ckpt = "flax-community/gpt2-rap-lyric-generator"
|
| 6 |
tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
|
| 7 |
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-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)
|