Update app.py

app.py

@@ -185,8 +185,13 @@ if uploaded_file is not None:
               t5=[]  
               t5_words = list(set(sent.split("\n")))
               for i in t5_words:
-                  t5+=i.split(" ")
-              t5_words=set(t5).difference(set(x.split(" ")))        
+                  for j in i.split(" "):
+                      t5+=j.split(',')
+              x1 = x.split(' ')    
+              e=[]
+              for i in x1:
+                 e+=i.split(',')                     
+              t5_words=set(t5).difference(set(e))      
               entities,words_out = extract_entities(sent)
               words_out+=t5_words
               bert_words = words_red_bert(sent)