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)
|