samueldomdey commited on
Commit
521307e
·
1 Parent(s): 267287b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -87,11 +87,11 @@ def bulk_function(filename):
87
  surprise.append(round(temp[i][6], 2))
88
 
89
  # define df
90
- df = pd.DataFrame(list(zip(ids,lines_s,preds,labels,scores_rounded, anger, disgust, fear, joy, neutral, sadness, surprise)), columns=[df_input.columns[0], df_input.columns[1],'pred','label','score', 'anger', 'disgust', 'fear', 'joy', 'neutral', 'sadness', 'surprise'])
91
  print(df)
92
  # save results to csv
93
  YOUR_FILENAME = filename.name.split(".")[0] + "_emotion_predictions" + ".csv" # name your output file
94
- df.to_csv(YOUR_FILENAME)
95
 
96
  # return dataframe for space output
97
  return YOUR_FILENAME
 
87
  surprise.append(round(temp[i][6], 2))
88
 
89
  # define df
90
+ df = pd.DataFrame(list(zip(ids,lines_s,preds,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'])
91
  print(df)
92
  # save results to csv
93
  YOUR_FILENAME = filename.name.split(".")[0] + "_emotion_predictions" + ".csv" # name your output file
94
+ df.to_csv(YOUR_FILENAME, index=False)
95
 
96
  # return dataframe for space output
97
  return YOUR_FILENAME