Update WNgen.py

WNgen.py

@@ -187,6 +187,7 @@ def wordnet_df(word,POS=False,seed_definition=None):
     df = pd.DataFrame(df) 
     df = df.drop_duplicates().reset_index()
     df = df.drop("index", axis=1)
+    df = df.convert_dtypes()
     return df
 
 def eval_pred_test(text, return_all = False):
@@ -292,6 +293,7 @@ def wordnet_parallel_df(word,seed_definition=None):
     df = pd.DataFrame(df) 
     df = df.drop_duplicates("Words").reset_index()
     df = df.drop("index", axis=1)
+    df = df.convert_dtypes()
     return df
 
 #@st.experimental_singleton(suppress_st_warning=True)
@@ -311,4 +313,5 @@ def cf_from_wordnet_df(seed,text,seed_definition=False):
     df["pred"] = df.text.apply(eval_pred_test)
     # added this because I think it will make the end results better if we ensure the seed is in the data we generate counterfactuals from.
     df['seed'] = df.Words.apply(lambda x: 'seed' if x.lower() == seed.lower() else 'alternative')
+    df = df.convert_dtypes()
     return df