EasyPhoto add s3 dowm

Lassi-Ki · Feb 5, 2024 · 3be1ade · 3be1ade
1 parent 264e4bd
commit 3be1ade
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/modules/api/api.py b/modules/api/api.py
@@ -934,12 +934,14 @@ def invocations(self, req: models.InvocationsRequest):
                         reload_vae_weights()
 
                 # Train
+                user_path = f'./datasets/{req.id}'
                 if req.s3Url !='':
-                    shared.download_dataset_from_s3(req.s3Url, f'./datasets/{req.id}')
+                    shared.download_dataset_from_s3(req.s3Url, user_path)
 
-                img_list = f'./datasets/{req.id}'
+                img_list = os.listdir(user_path)
                 encoded_images = []
                 for idx, img_path in enumerate(img_list):
+                    img_path = os.path.join(user_path, img_path)
                     with open(img_path, "rb") as f:
                         encoded_image = base64.b64encode(f.read()).decode("utf-8")
                         encoded_images.append(encoded_image)