pipeline update

app.py

@@ -2,7 +2,7 @@ import pandas as pd
 import streamlit as st
 from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
 
-from src import GBRT, wikipedia_search, wikidata_search, google_search
+from src import GBRT, wikidata_search, google_search
 
 TYPE = {
     'LOC': ' location',
@@ -53,9 +53,8 @@ def get_candidates(mentions_tags):
             candidates.append((mention, cache[(mention, tag)]))
         else:
             cands = wikidata_search(mention, limit=3)
-            cands = list(set(cands + google_search(mention, 1)))
-            if cands == []:
-                cands = wikipedia_search(mention, limit=3)
+            if cands == [] and tag != 'PER':
+                cands = google_search(mention, limit=3)
             cache[(mention, tag)] = cands
             candidates.append((mention, cands))
             print(mention, cands)