Refinement
This commit is contained in:
@@ -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)
|
||||
|
Reference in New Issue
Block a user