Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -110,7 +110,9 @@ def run(bucket_folder, person_name, req_id):
|
|
| 110 |
for _ in range(n_samples):
|
| 111 |
img = pipeline.inference(prompt, (1, H, W), neg_prompt, id_embeddings, id_scale, scale, steps)[0]
|
| 112 |
ims.append(np.array(img))
|
| 113 |
-
|
|
|
|
|
|
|
| 114 |
file_paths = []
|
| 115 |
for i, img in enumerate(ims):
|
| 116 |
if isinstance(img, torch.Tensor):
|
|
@@ -121,6 +123,7 @@ def run(bucket_folder, person_name, req_id):
|
|
| 121 |
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
|
| 122 |
img.save(temp_file.name)
|
| 123 |
file_paths.append(temp_file.name)
|
|
|
|
| 124 |
|
| 125 |
# Upload images to S3
|
| 126 |
OUTPUT_BUCKET_FOLDER = os.path.join(os.path.dirname(os.path.dirname(bucket_folder)), 'output') #'user_id/request_id/output/'
|
|
|
|
| 110 |
for _ in range(n_samples):
|
| 111 |
img = pipeline.inference(prompt, (1, H, W), neg_prompt, id_embeddings, id_scale, scale, steps)[0]
|
| 112 |
ims.append(np.array(img))
|
| 113 |
+
print('IMS is ', ims)
|
| 114 |
+
print('Length of ims is ',len(ims))
|
| 115 |
+
|
| 116 |
file_paths = []
|
| 117 |
for i, img in enumerate(ims):
|
| 118 |
if isinstance(img, torch.Tensor):
|
|
|
|
| 123 |
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
|
| 124 |
img.save(temp_file.name)
|
| 125 |
file_paths.append(temp_file.name)
|
| 126 |
+
print('File paths are ',file_paths)
|
| 127 |
|
| 128 |
# Upload images to S3
|
| 129 |
OUTPUT_BUCKET_FOLDER = os.path.join(os.path.dirname(os.path.dirname(bucket_folder)), 'output') #'user_id/request_id/output/'
|