Update app.py

app.py

@@ -321,8 +321,8 @@ def extract_relevant_portions(document_texts, query, max_portions=3, portion_siz
         sentences = nltk.sent_tokenize(doc_text)  # Split document into sentences
         doc_relevant_portions = []
         # Extract entities from the entire document
-        ner_biobert = models['ner_pipeline']
-        doc_entities = extract_entities(doc_text, ner_biobert)
+        #ner_biobert = models['ner_pipeline']
+        doc_entities = extract_entities(doc_text)
         print(f"Document {doc_id} Entities: {doc_entities}")
         for i, sentence in enumerate(sentences):
             # Extract entities from the sentence