Spaces:

flax-community
/

roberta-indonesian

Runtime error

w11wo commited on Jul 17, 2021

Commit

31d37bb

•

1 Parent(s): 1ae1bf6

fixed scalar value error

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ def main():
         else:
             # render masked language modeling table
             mlm_result = mask_api(inputs=user_input)
-            st.write(mlm_result)
             mlm_df = pd.DataFrame(mlm_result)
             mlm_df.drop(columns=["token", "token_str"], inplace=True)
             mlm_df_styled = mlm_df.copy(deep=False)

         else:
             # render masked language modeling table
             mlm_result = mask_api(inputs=user_input)
+            # st.write(mlm_result)
             mlm_df = pd.DataFrame(mlm_result)
             mlm_df.drop(columns=["token", "token_str"], inplace=True)
             mlm_df_styled = mlm_df.copy(deep=False)