From bd407e51f9453b639303f67fd16f320f0aa9d358 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Mon, 4 Mar 2024 14:58:01 -0500 Subject: [PATCH] code clean up --- backend/run_evals.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/run_evals.py b/backend/run_evals.py index 75e76eb..a5cfefb 100644 --- a/backend/run_evals.py +++ b/backend/run_evals.py @@ -11,6 +11,8 @@ from evals.config import EVALS_DIR from evals.core import generate_code_core from evals.utils import image_to_data_url +STACK = "html_tailwind" + async def main(): INPUT_DIR = EVALS_DIR + "/inputs" @@ -23,7 +25,7 @@ async def main(): for filename in evals: filepath = os.path.join(INPUT_DIR, filename) data_url = await image_to_data_url(filepath) - task = generate_code_core(data_url, "html_tailwind") + task = generate_code_core(data_url, STACK) tasks.append(task) results = await asyncio.gather(*tasks)