thechaiexperiment commited on
Commit
6a8df74
·
1 Parent(s): 5bb49bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -358,6 +358,7 @@ def remove_duplicates(selected_parts):
358
 
359
  def extract_entities(text):
360
  try:
 
361
  inputs = biobert_tokenizer(text, return_tensors="pt")
362
  outputs = biobert_model(**inputs)
363
  predictions = torch.argmax(outputs.logits, dim=2)
 
358
 
359
  def extract_entities(text):
360
  try:
361
+ biobert_tokenizer = models['bio_tokenizer']
362
  inputs = biobert_tokenizer(text, return_tensors="pt")
363
  outputs = biobert_model(**inputs)
364
  predictions = torch.argmax(outputs.logits, dim=2)