DINGOLANI commited on
Commit
9602f27
·
verified ·
1 Parent(s): d3b176d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -32,7 +32,10 @@ def parse_fashion_query(query):
32
  for label in result:
33
  entity = label["entity"]
34
  word = label["word"]
35
- readable_entity = label_map.get(entity, "Unknown")
 
 
 
36
 
37
  # Merge subwords (handling "##tokens")
38
  if word.startswith("##") and prev_entity == readable_entity:
 
32
  for label in result:
33
  entity = label["entity"]
34
  word = label["word"]
35
+ readable_entity = label_map.get(entity)
36
+
37
+ if not readable_entity:
38
+ continue # Skip unknown labels
39
 
40
  # Merge subwords (handling "##tokens")
41
  if word.startswith("##") and prev_entity == readable_entity: