Removed unnecessary parameter

This commit is contained in:
2025-09-03 11:42:10 -07:00
parent d330f47462
commit 184edcb7e4
2 changed files with 10 additions and 10 deletions

View File

@@ -3,7 +3,6 @@ import sys
import csv
import logging
from sentence_transformers import SentenceTransformer
import tqdm
import click
from tabulate import tabulate, SEPARATING_LINE
@@ -59,8 +58,7 @@ def recommend(text, paths, model, interactive, skip_ucs):
the synonyms an 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)
ctx = InferenceContext(model)
if text is not None:
print_recommendation(None, text, ctx, interactive_rename=False)
@@ -177,8 +175,7 @@ def evaluate(dataset, offset, limit, model, no_foley):
classified according to their subject and not wether or not they were
foley, and so these categories can be excluded with the --no-foley option.
"""
m = SentenceTransformer(model)
ctx = InferenceContext(m, model)
ctx = InferenceContext(model)
reader = csv.reader(dataset)
print(f"Evaluating model {model}...")