Update app.py

app.py

@@ -3,7 +3,24 @@ import os
 from moviepy.editor import *
 import gradio as gr
 import re
+import nltk
+from nltk.stem import WordNetLemmatizer
 
+
+
+
+def get_lemma(word):
+    nltk.download('wordnet')
+    lemmatizer = WordNetLemmatizer()
+    lemma = lemmatizer.lemmatize(word)
+    return lemma
+
+def apply_lemma_to_list(word_list):
+    updated_list = []
+    for word in word_list:
+        updated_list.append(get_lemma(word))
+    return updated_list
+    
 def parse_string(string, dataset):
     parsed_list = []
     start = 0
@@ -67,8 +84,10 @@ def texttoSign(text):
       for word in listofwords:
         if word not in VideosNames:
           listofwords[index]=(list(word))
+          
         index+=1
       listofwords=flatten_lists(listofwords)
+      listofwords=apply_lemma_to_list(listofwords)
       clips=[]
       for i in range(len(listofwords)):