bharat-raghunathan commited on
Commit
9429ed6
1 Parent(s): 165591c

Update app v0.2

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -7,6 +7,9 @@ def lyrics_categories(input_text):
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..."),
 
7
  tokenizer = AutoTokenizer.from_pretrained(spotify_model, use_auth_token=True)
8
  inputs = tokenizer(input_text, return_tensors="pt")
9
  outputs = model(**inputs)
10
+ return outputs
11
+
12
+ return clean_outputs
13
 
14
  iface = gr.Interface(fn=lyrics_categories,
15
  inputs=gr.inputs.Textbox(lines=7, placeholder="Enter song lyrics here..."),