Spaces:

bharat-raghunathan
/

song-lyrics-classifier

Runtime error

bharat-raghunathan commited on Mar 30, 2023

Commit

a7b965a

•

1 Parent(s): b0f5706

Update link to model (app v1.0b3)

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 from examples import yellow, stairway, numb, puppets, firework
 def lyrics_categories(input_text):
-  spotify_model = "juliensimon/autonlp-huggingface-demo-song-lyrics-18923587"
   model = AutoModelForSequenceClassification.from_pretrained(spotify_model)
   tokenizer = AutoTokenizer.from_pretrained(spotify_model)
   labels = model.config.id2label

 from examples import yellow, stairway, numb, puppets, firework
 def lyrics_categories(input_text):
+  spotify_model = "juliensimon/autonlp-song-lyrics-18753417"
   model = AutoModelForSequenceClassification.from_pretrained(spotify_model)
   tokenizer = AutoTokenizer.from_pretrained(spotify_model)
   labels = model.config.id2label