Update app.py

app.py

@@ -67,13 +67,6 @@ def main():
     mlm_model = "BigSalmon/BestMask2"
     mask_api = InferenceApi(mlm_model)
 
-    if analyze == "Emotion":
-        sa_model = "StevenLimcorn/indonesian-roberta-base-emotion-classifier"
-    elif analyze == "Sentiment":
-        sa_model = "w11wo/indonesian-roberta-base-sentiment-classifier"
-
-    sa_pipeline = pipeline("sentiment-analysis", model=sa_model, tokenizer=sa_model)
-
     if len(user_input) > 0:
         try:
             user_input.index(MASK_TOKEN)
@@ -97,20 +90,6 @@ def main():
             )
             display_table(mlm_df_styled, "🎈 Top 5 Predictions")
 
-            # render sentiment analysis table
-            sa_df = pd.DataFrame(columns=["sequence", "label", "score"])
-            for sequence in mlm_df["sequence"].values:
-                sa_output = sa_pipeline(sequence)  # predict for every mlm output
-                result_dict = {"sequence": sequence}
-                result_dict.update(sa_output[0])
-                sa_df = sa_df.append(result_dict, ignore_index=True)
-
-            sa_df["label"] = sa_df["label"].apply(lambda x: x + " " + EMOJI_MAP[x])
-            sa_df_styled = sa_df.copy(deep=False)
-            sa_df_styled = sa_df_styled.style.set_properties(
-                subset=["sequence", "label", "score"], **{"text-align": "left"}
-            )
-            display_table(sa_df_styled, "🤔 By saying that, I guess you are feeling..")
 
 
 if __name__ == "__main__":