Update app.py

app.py

@@ -72,7 +72,7 @@ def bulk_function(filename):
   scores = (np.exp(predictions[0])/np.exp(predictions[0]).sum(-1,keepdims=True)).max(1)
 
   # round scores
-  scores_rounded = [round(score, 2) for score in scores]
+  scores_rounded = [round(score, 3) for score in scores]
 
     # scores raw
   temp = (np.exp(predictions[0])/np.exp(predictions[0]).sum(-1,keepdims=True))
@@ -97,7 +97,7 @@ def bulk_function(filename):
     surprise.append(round(temp[i][6], 3))
 
   # define df
-  df = pd.DataFrame(list(zip(ids,lines_s,labels,scores_rounded, anger, disgust, fear, joy, neutral, sadness, surprise)), columns=[df_input.columns[0], df_input.columns[1],'label','score', 'anger', 'disgust', 'fear', 'joy', 'neutral', 'sadness', 'surprise'])
+  df = pd.DataFrame(list(zip(ids,lines_s,labels,scores_rounded, anger, disgust, fear, joy, neutral, sadness, surprise)), columns=[df_input.columns[0], df_input.columns[1],'max_label','max_score', 'anger', 'disgust', 'fear', 'joy', 'neutral', 'sadness', 'surprise'])
   print(df)
   # save results to csv
   YOUR_FILENAME = filename.name.split(".")[0] + "_emotion_predictions" + ".csv"  # name your output file