def infer_topics()

in src/kg.py [0:0]

• 13 lines of code
• 6 McCabe index (conditional complexity)

def infer_topics(queries, pbar=True):
    topics = []
    if pbar:
        for query in tqdm(queries):
            entities = gliner_model.predict_entities(query, labels, threshold=0.3)
            themes = list({entity["label"] for entity in entities})
            topics.append(themes)
    else:
        for query in queries:
            entities = gliner_model.predict_entities(query, labels, threshold=0.3)
            themes = list({entity["label"] for entity in entities})
            topics.append(themes)
    return topics