Spaces:
Running
Running
P01yH3dr0n
commited on
Commit
•
cb7ba0d
1
Parent(s):
d907127
Update utils.py
Browse files
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') == '
|
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()
|