Minor Bug fix

app.py

@@ -225,11 +225,11 @@ def _predict_sentiment(p):
   inputs = sentiment.tokenizer(p, return_tensors="pt")
   # Pass inputs through model
   outputs = sentiment.model(**inputs)
-  sentiment_map = sentiment.utility.capitalize_first_letter(sentiment.model.config.label2id.keys())
+  # sentiment_map = sentiment.utility.capitalize_first_letter(sentiment.model.config.label2id.keys())
   out_data = outputs[0][0]
   scores = out_data.detach().numpy()
   scores = scipy.special.softmax(scores)
-  # sentiment_map = ['Sadness', 'Joy', 'Love', 'Anger', 'Fear' , "Surprise"]
+  sentiment_map = ['Sadness', 'Joy', 'Love', 'Anger', 'Fear' , "Surprise"]
   df_out = pandas.DataFrame([scores], columns=sentiment_map)
   df_out = df_out[['Love' , 'Joy', 'Surprise' , 'Fear', 'Sadness', 'Anger']]
   return df_out