Install commit
This commit is contained in:
@@ -89,6 +89,9 @@ def run_inference(batch_file_path: str, model_list_path: str):
|
||||
df = pd.read_csv(batch_file_path, encoding="utf-8")
|
||||
if "input" not in df.columns:
|
||||
raise ValueError("The input CSV must contain a column named 'input'.")
|
||||
|
||||
# 에러 발생한 행 저장용 DataFrame 초기화
|
||||
error_rows = pd.DataFrame(columns=df.columns)
|
||||
|
||||
# 추론 수행
|
||||
for model in model_list:
|
||||
@@ -166,7 +169,7 @@ def merge_results():
|
||||
def download_latest_file():
|
||||
try:
|
||||
# processed 디렉토리 경로
|
||||
directory = "LLM_asyncio/processed"
|
||||
directory = "processed"
|
||||
|
||||
csv_files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith(".csv")]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user