Spaces:
Sleeping
Sleeping
Commit
·
6a8df74
1
Parent(s):
5bb49bd
Update app.py
Browse files
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)
|