Removed unnecessary parameter
This commit is contained in:
@@ -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}...")
|
||||
|
Reference in New Issue
Block a user