Update app.py

app.py

@@ -359,6 +359,7 @@ def remove_duplicates(selected_parts):
 def extract_entities(text):
     try:
         biobert_tokenizer = models['bio_tokenizer']
+        biobert_model = models['bio_model']
         inputs = biobert_tokenizer(text, return_tensors="pt")
         outputs = biobert_model(**inputs)
         predictions = torch.argmax(outputs.logits, dim=2)