bharat-raghunathan commited on
Commit
165591c
1 Parent(s): 2652a5e

Create app v0.1

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
+
4
+ def lyrics_categories(input_text):
5
+ spotify_model = "spotify/autonlp-huggingface-demo-song-lyrics-18923587"
6
+ model = AutoModelForSequenceClassification.from_pretrained(spotify_model, use_auth_token=True)
7
+ tokenizer = AutoTokenizer.from_pretrained(spotify_model, use_auth_token=True)
8
+ inputs = tokenizer(input_text, return_tensors="pt")
9
+ outputs = model(**inputs)
10
+
11
+ iface = gr.Interface(fn=lyrics_categories,
12
+ inputs=gr.inputs.Textbox(lines=7, placeholder="Enter song lyrics here..."),
13
+ outputs="text")
14
+ iface.launch()