lalalalalalalalalala
commited on
Commit
β’
0228656
1
Parent(s):
bca51b0
Update run.py
Browse files
run.py
CHANGED
@@ -20,6 +20,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
|
|
20 |
progress_info = []
|
21 |
with tempfile.TemporaryDirectory() as temp_dir:
|
22 |
csv_filename = os.path.join(temp_dir, str(parquet_index) + '_caption.csv')
|
|
|
23 |
with open(csv_filename, mode='w', newline='') as csv_file:
|
24 |
fieldnames = ['md5', 'caption']
|
25 |
writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
|
|
|
20 |
progress_info = []
|
21 |
with tempfile.TemporaryDirectory() as temp_dir:
|
22 |
csv_filename = os.path.join(temp_dir, str(parquet_index) + '_caption.csv')
|
23 |
+
print(csv_filename)
|
24 |
with open(csv_filename, mode='w', newline='') as csv_file:
|
25 |
fieldnames = ['md5', 'caption']
|
26 |
writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
|