P01yH3dr0n commited on
Commit
046c320
1 Parent(s): 05346f6

img from bytes

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -172,11 +172,11 @@ def augment_image(image, width, height, req_type, selection, factor=1, defry=0,
172
  zipfile_in_memory = io.BytesIO(response.content)
173
  with zipfile.ZipFile(zipfile_in_memory, 'r') as zip_ref:
174
  if len(zip_ref.namelist()):
175
- files = []
176
  for file_name in zip_ref.namelist():
177
  with zip_ref.open(file_name) as file:
178
- files.append(file.read())
179
- return files
180
  else:
181
  messages = json.loads(response.content)
182
  raise gr.Error(messages["statusCode"] + ": " + messages["message"])
 
172
  zipfile_in_memory = io.BytesIO(response.content)
173
  with zipfile.ZipFile(zipfile_in_memory, 'r') as zip_ref:
174
  if len(zip_ref.namelist()):
175
+ images = []
176
  for file_name in zip_ref.namelist():
177
  with zip_ref.open(file_name) as file:
178
+ images.append(image_from_bytes(file.read()))
179
+ return images
180
  else:
181
  messages = json.loads(response.content)
182
  raise gr.Error(messages["statusCode"] + ": " + messages["message"])