w11wo commited on
Commit
ae2fa97
Β·
1 Parent(s): dc4fa76

fixed flushing df

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -66,7 +66,8 @@ def main():
66
  mlm_result = mask_api(inputs=user_input)
67
  mlm_df = pd.DataFrame(mlm_result)
68
  mlm_df.drop(columns=["token", "token_str"], inplace=True)
69
- mlm_df_styled = mlm_df.style.set_properties(
 
70
  subset=["sequence", "score"], **{"text-align": "left"}
71
  )
72
  display_table(mlm_df_styled, "🎈 Top 5 Predictions")
@@ -80,7 +81,8 @@ def main():
80
  sa_df = sa_df.append(result_dict, ignore_index=True)
81
 
82
  sa_df["label"] = sa_df["label"].apply(lambda x: x + " " + EMOTION_MAP[x])
83
- sa_df_styled = sa_df.style.set_properties(
 
84
  subset=["sequence", "label", "score"], **{"text-align": "left"}
85
  )
86
  display_table(sa_df_styled, "πŸ€” By saying that, I guess you are feeling..")
 
66
  mlm_result = mask_api(inputs=user_input)
67
  mlm_df = pd.DataFrame(mlm_result)
68
  mlm_df.drop(columns=["token", "token_str"], inplace=True)
69
+ mlm_df_styled = mlm_df.copy(deep=False)
70
+ mlm_df_styled = mlm_df_styled.style.set_properties(
71
  subset=["sequence", "score"], **{"text-align": "left"}
72
  )
73
  display_table(mlm_df_styled, "🎈 Top 5 Predictions")
 
81
  sa_df = sa_df.append(result_dict, ignore_index=True)
82
 
83
  sa_df["label"] = sa_df["label"].apply(lambda x: x + " " + EMOTION_MAP[x])
84
+ sa_df_styled = sa_df.copy(deep=False)
85
+ sa_df_styled = sa_df_styled.style.set_properties(
86
  subset=["sequence", "label", "score"], **{"text-align": "left"}
87
  )
88
  display_table(sa_df_styled, "πŸ€” By saying that, I guess you are feeling..")