Refinement

This commit is contained in:
Jamie Hardt
2025-08-27 15:47:38 -07:00
parent 65bdf3256e
commit 4a5397f153

View File

@@ -2,8 +2,6 @@ import os
import sys
import csv
from typing import Optional
from sentence_transformers import SentenceTransformer
import tqdm
import click
@@ -51,6 +49,11 @@ def recommend(text, paths, model):
"""
Infer a UCS category for a text description
"Description" text metadata is extracted from audio files given as PATHS,
or text can be provided directly using the "--text" option. The selected
model is then used to attempt to classify the given text according to
the synonyms and explanations definied for each UCS subcategory. A list
of ranked subcategories is printed to the terminal for each PATH.
"""
m = SentenceTransformer(model)
ctx = InferenceContext(m, model)