Spaces:

dell-research-harvard
/

newsdejavu

Sleeping

tombryan commited on Apr 19

Commit

01ef847

•

1 Parent(s): a84190a

Error fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ def query(sentence: str) -> List[str]:
     # Send embedding to Azure VM
     response = requests.post(f"http://{AZURE_VM_ALABAMA}/retrieve", json = req)
     doc = response.json()
-    article = doc['bboxes'][doc['article_id']]
     return article['raw_text']

     # Send embedding to Azure VM
     response = requests.post(f"http://{AZURE_VM_ALABAMA}/retrieve", json = req)
     doc = response.json()
+    article = doc['bboxes'][int(doc['article_id'])]
     return article['raw_text']