Install commit

This commit is contained in:
b24503@hanmaceng.co.kr
2025-01-07 12:13:09 +09:00
parent fc3ead893a
commit a9628b9013
9 changed files with 93 additions and 2 deletions

View File

@@ -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")]