Update app.py

app.py

@@ -64,7 +64,7 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
             try:
                 res.append(voc.index[voc['word'] == word].values[0])
             except (IndexError, KeyError) as e:
-                st.write('Token Unfound')
+                st.write('Token not found')
                 continue
         return res
     
@@ -75,8 +75,8 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
         for e in id_list:
             try:
                 res.extend(list(df.loc[e]['ID2'].values)) 
-            except KeyError:
-                st.write('Token Unfound')
+            except (KeyError, AttributeError) as f:
+                st.write('Token not found')
                 continue
         return list(set(res))