Update app.py

app.py

@@ -26,13 +26,14 @@ def fix_words(words: typing.List[str]) -> typing.List[str]:
         fixed_words = []
 
         for word, pos in tagger.tag(words):
-            if pos[-1] == "e":
+            if pos[-1] == "EZ":
                 if word[-1] != "ِ":
                     if (word[-1] == "ه") and (word[-2] != "ا"):
                         word += "‌ی"
-                    word += "ِ"
+                word += "ِ"
+                    
 
-            fixed_words.append(pos[1])
+            fixed_words.append(word)
 
         return fixed_words
         #return tagger.tag(words)