Revert back to v0.91

app.py

@@ -13,7 +13,7 @@ def lyrics_categories(input_text):
   predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
   predictions = predictions.detach().numpy()[0]
   index_sorted = np.argsort(predictions)[::-1]
-  clean_outputs = {labels[idx]:predictions[idx] for idx in index_sorted}
+  clean_outputs = {labels[idx]:str(predictions[idx]) for idx in index_sorted}
   print(clean_outputs)
   return clean_outputs