Update app.py

app.py

@@ -26,7 +26,7 @@ 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 += "‌ی"
@@ -34,8 +34,8 @@ def fix_words(words: typing.List[str]) -> typing.List[str]:
 
             fixed_words.append(word)
 
-        #return fixed_words
-        return tagger.tag(words)
+        return fixed_words
+        #return tagger.tag(words)
     
 iface = gr.Interface(fn=preprocess_text, inputs="text", outputs="text")
 iface.launch()