srijaydeshpande commited on
Commit
eeaeb90
·
verified ·
1 Parent(s): c549417

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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/'