Track the app file

app.py

@@ -60,7 +60,7 @@ dataset = datasets.load_dataset(dataset_name, token=os.environ["HF_TOKEN"])["tes
 
 for sample in dataset:
     text = sample["sentence"]
-    labels= [DIALECTS[i] for i in range(len(DIALECTS)) if int(sample[DIALECTS[i]]) == 1]
+    labels= [DIALECTS[i] for i in range(len(DIALECTS)) if DIALECTS[i] in sample.keys() and int(sample[DIALECTS[i]]) == 1]
     pred = predict_top_p(text)
     sample["pred"] = pred
     st.write("Text:", text)