Fixing dbpedia

app.py

@@ -19,6 +19,7 @@ for k, v in os.environ.items():
 
 AVAILABLE_YEARS = ['All Years']
 
+REQUEST_HEADERS = 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Ubuntu Chromium/37.0.2062.94 Chrome/37.0.2062.94 Safari/537.36'
 
 
 def find_sep_token(tokenizer):
@@ -58,6 +59,9 @@ if gr.NO_RELOAD:
     embed_mask_tok = find_mask_token(embedding_tokenizer)
     embed_sep_tok = find_sep_token(embedding_tokenizer)
 
+    img_download_session = requests.Session()
+
+
     # with open(REF_INDEX_PATH, 'r') as f:
     #     news_paths = [l.strip() for l in f.readlines()]
 
@@ -140,10 +144,13 @@ def query(sentence: str, state: str, years: List[str]) -> List[str]:
 
     doc = response.json()
     article = doc['bboxes'][int(doc['article_id'])]
-    if len(article['lccn']['dpedia_ids']) == 0:
+    if len(doc['lccn']['dbpedia_ids']) == 0:
         location = 'Unknown'
     else:
-        location = doc['lccn']['dpedia_ids'][0].replace('%2C_', ', ')
+        location = doc['lccn']['dbpedia_ids'][0].replace('%2C_', ', ')
+    
+    # response = img_download_session.get(ca_url, headers = {'User-Agent': random.choice(USER_HEADERS)})
+
     results = {
         'newspaper_name': doc['lccn']['title'],
         'location': location,