P01yH3dr0n commited on
Commit
cb7ba0d
1 Parent(s): d907127

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -105,7 +105,7 @@ def generate_novelai_image(
105
  return None, {'message': 'NAI response timeout'}
106
 
107
  # Process the response
108
- if response.headers.get('Content-Type') == 'application/x-zip-compressed':
109
  zipfile_in_memory = io.BytesIO(response.content)
110
  with zipfile.ZipFile(zipfile_in_memory, 'r') as zip_ref:
111
  file_names = zip_ref.namelist()
 
105
  return None, {'message': 'NAI response timeout'}
106
 
107
  # Process the response
108
+ if response.headers.get('Content-Type') == 'binary/octet-stream':
109
  zipfile_in_memory = io.BytesIO(response.content)
110
  with zipfile.ZipFile(zipfile_in_memory, 'r') as zip_ref:
111
  file_names = zip_ref.namelist()