diff --git a/ucsinfer/__main__.py b/ucsinfer/__main__.py index 3e418bd..4b2306e 100644 --- a/ucsinfer/__main__.py +++ b/ucsinfer/__main__.py @@ -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)