Update app.py

app.py

@@ -4,12 +4,12 @@ import pandas as pd
 # Load and parse the CSV file from Hugging Face
 def load_data():
     url = "https://huggingface.co/datasets/unijoh/RAVNlex/resolve/main/RAVNlex.csv"
-    df = pd.read_csv(url, delimiter='\t', encoding='iso-8859-10', names=["#ORTO", "#PPOS", "#PHON1", "#PHON2", "#COMM"])
+    df = pd.read_csv(url, delimiter='\t', encoding='iso-8859-10', names=["#ORTO", "#PPOS", "#PHON1", "#PHON2", "#COMM"], dtype=str)
     lemmas = {}
     current_lemma = None
     
     for _, row in df.iterrows():
-        if row['#ORTO'] == '---':
+        if pd.isna(row['#ORTO']) or row['#ORTO'] == '---':
             current_lemma = None
         elif current_lemma is None:
             current_lemma = row['#ORTO'].replace("ORTO:", "")