Upload 17 files

utils/entity_extraction.py

@@ -79,7 +79,8 @@ def all_keywords_combs(texts):
 
 
 def extract_keywords(query_text, model):
-    prompt = f"###Instruction:Extract the important keywords which describe the context accurately.\n\nInput:{query_text}\n\n###Response:"
+    prompt = "###Instruction: Identify the key entities that accurately describe the context.\n\nInput:{query_text}\n\n###Response:"
+    #prompt = f"###Instruction:Extract the important keywords which describe the context accurately.\n\nInput:{query_text}\n\n###Response:"
     response = model.predict(prompt)
     keywords = response.split(", ")
     keywords = keywords_no_companies(keywords)