Update app v0.5

app.py

@@ -12,11 +12,12 @@ def lyrics_categories(input_text):
   predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
   predictions = predictions.detach().numpy()[0]
   predictions = predictions * 100
-  index_sorted = np.argsort(predictions)
+  index_sorted = np.argsort(predictions)[::-1]
   clean_outputs = {label[idx]:predictions[idx] for idx in index_sorted}
+  print(clean_outputs)
   return clean_outputs
  
 iface = gr.Interface(fn=lyrics_categories, 
-                     inputs=gr.inputs.Textbox(lines=7, placeholder="Enter song lyrics here..."),
-                     outputs=gr.outputs.Label(num_top_classes=5, label="Lyric Categories"))
+                     inputs=gr.inputs.Textbox(lines=20, placeholder="Enter song lyrics here...", label="Song Lyrics"),
+                     outputs=gr.outputs.Label(num_top_classes=5, label="Lyrics Categories"))
 iface.launch()