Commit
•
b0f5706
1
Parent(s):
f3d107d
Update model link (app to v1.0b2)
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
|
| 5 |
from examples import yellow, stairway, numb, puppets, firework
|
| 6 |
|
| 7 |
def lyrics_categories(input_text):
|
| 8 |
-
spotify_model = "
|
| 9 |
model = AutoModelForSequenceClassification.from_pretrained(spotify_model)
|
| 10 |
tokenizer = AutoTokenizer.from_pretrained(spotify_model)
|
| 11 |
labels = model.config.id2label
|
|
|
|
| 5 |
from examples import yellow, stairway, numb, puppets, firework
|
| 6 |
|
| 7 |
def lyrics_categories(input_text):
|
| 8 |
+
spotify_model = "juliensimon/autonlp-huggingface-demo-song-lyrics-18923587"
|
| 9 |
model = AutoModelForSequenceClassification.from_pretrained(spotify_model)
|
| 10 |
tokenizer = AutoTokenizer.from_pretrained(spotify_model)
|
| 11 |
labels = model.config.id2label
|