P01yH3dr0n commited on
Commit
a3512f8
1 Parent(s): 399c110

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -99,7 +99,7 @@ def generate_novelai_image(
99
  payload['parameters']['strength'] = i2i_str
100
  payload['parameters']['extra_noise_seed'] = seed
101
  # Send the POST request
102
- response = requests.post(url, json=payload, headers=headers)
103
 
104
  # Process the response
105
  if response.headers.get('Content-Type') == 'application/x-zip-compressed':
 
99
  payload['parameters']['strength'] = i2i_str
100
  payload['parameters']['extra_noise_seed'] = seed
101
  # Send the POST request
102
+ response = requests.post(url, json=payload, headers=headers, timeout=30)
103
 
104
  # Process the response
105
  if response.headers.get('Content-Type') == 'application/x-zip-compressed':