fix: upload multiple images

runpod-workers · May 17, 2023 · b6bd813 · b6bd813
1 parent 9b59658
commit b6bd813
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/rp_handler.py b/src/rp_handler.py
@@ -57,7 +57,8 @@ def run(job):
 
     job_output = []
     for index, img_path in enumerate(img_paths):
-        image_url = upload_file_to_bucket(job['id'], img_path, index)
+        file_name = f"{job['id']}_{index}.png"
+        image_url = upload_file_to_bucket(file_name, img_path)
 
         job_output.append({
             "image": image_url,