code clean up

This commit is contained in:
Abi Raja 2024-03-04 14:58:01 -05:00
parent 5e3a174203
commit bd407e51f9

View File

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