hyunjunian commited on
Commit
cf44933
·
1 Parent(s): 74c545b
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -26,8 +26,7 @@ def extract_pii(text):
26
  },
27
  )
28
  pii_entities = res.json()["results"]["documents"][0]["entities"]
29
- return map(lambda row: [row["text"], row["category"], row["confidenceScore"]], pii_entities)
30
-
31
 
32
  demo = gr.Interface(extract_pii, inputs="textbox", outputs="dataframe")
33
 
 
26
  },
27
  )
28
  pii_entities = res.json()["results"]["documents"][0]["entities"]
29
+ return list(map(lambda row: [row["text"], row["category"], row["confidenceScore"]], pii_entities))
 
30
 
31
  demo = gr.Interface(extract_pii, inputs="textbox", outputs="dataframe")
32