Update app.py

app.py

@@ -101,11 +101,12 @@ def find_query(query,sen,nouns):
             noun_list.append(n[0])
     nouns_l=list(nouns.keys())
     for nn in nouns_l:
-        if nn in noun_list:
-            if nn in noun_box:
-                noun_box[str(nn)].append(nouns[nn])    
-            else:
-                noun_box[str(nn)]=[nouns[nn]]           
+        for nl in noun_list:
+            if nl in nn:
+                if nl in noun_box:
+                    noun_box[str(nl)].append(nouns[nn])    
+                else:
+                    noun_box[str(nl)]=[nouns[nn]]           
     return noun_box
 
 with gr.Blocks() as app: