Update app.py

app.py

@@ -6,7 +6,7 @@ st.set_page_config(layout="wide")
 dataset = load_dataset("GroNLP/divemt")
 df = dataset["train"].to_pandas()
 unique_src = df[["item_id", "src_text"]].drop_duplicates(subset="item_id")
-langs = list(df["lang"].unique())
+langs = list(df["lang_id"].unique())
 
 st.title("DivEMT Explorer")
 
@@ -59,7 +59,7 @@ for lang in langs:
                 key=f"{lang}_tasks"
             )
         columns = st.columns(len(tasks))
-        lang_data = df[(df["item_id"] == unique_src.iloc[int(item_id)]["item_id"]) & (df["lang"] == lang)]
+        lang_data = df[(df["item_id"] == unique_src.iloc[int(item_id)]["item_id"]) & (df["lang_id"] == lang)]
         lang_dicts = lang_data.to_dict("records")
         ht = [x for x in lang_dicts if x["task_type"] == "ht"][0]
         pe1 = [x for x in lang_dicts if x["task_type"] == "pe1"][0]