atrytone commited on
Commit
540cb8a
Β·
1 Parent(s): cfca6ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -44,9 +44,12 @@ def inference(query, k=30):
44
  date = doc.metadata.get('date', 'None')
45
  link = doc.metadata.get('link', 'None')
46
  submitter = doc.metadata.get('submitter', 'None')
47
- journal = doc.metadata.get('journal', 'None').strip()
48
- if journal == '':
 
49
  journal = 'None'
 
 
50
  # For journals
51
  if journal not in j_bucket:
52
  j_bucket[journal] = score
 
44
  date = doc.metadata.get('date', 'None')
45
  link = doc.metadata.get('link', 'None')
46
  submitter = doc.metadata.get('submitter', 'None')
47
+ # journal = doc.metadata.get('journal', 'None').strip()
48
+ journal = doc.metadata['journal']
49
+ if (journal == None or journal.strip() == ''):
50
  journal = 'None'
51
+ else:
52
+ journal = journal.strip()
53
  # For journals
54
  if journal not in j_bucket:
55
  j_bucket[journal] = score