Spaces:

P01yH3dr0n
/

client

Running

P01yH3dr0n commited on Mar 22

Commit

a3512f8

•

1 Parent(s): 399c110

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -99,7 +99,7 @@ def generate_novelai_image(
         payload['parameters']['strength'] = i2i_str
         payload['parameters']['extra_noise_seed'] = seed
     # Send the POST request
-    response = requests.post(url, json=payload, headers=headers)
     # Process the response
     if response.headers.get('Content-Type') == 'application/x-zip-compressed':

         payload['parameters']['strength'] = i2i_str
         payload['parameters']['extra_noise_seed'] = seed
     # Send the POST request
+    response = requests.post(url, json=payload, headers=headers, timeout=30)
     # Process the response
     if response.headers.get('Content-Type') == 'application/x-zip-compressed':