impart-client

Paused

P01yH3dr0n commited on May 24

Commit

825d6b5

•

1 Parent(s): a3bc818

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -101,7 +101,7 @@ def generate_novelai_image(
         payload['parameters']['extra_noise_seed'] = seed
     # Send the POST request
     try:
-        response = requests.post(url, json=payload, headers=headers)
     except:
         return None, {'message': 'NAI response timeout'}

         payload['parameters']['extra_noise_seed'] = seed
     # Send the POST request
     try:
+        response = requests.post(url, json=payload, headers=headers, timeout=180)
     except:
         return None, {'message': 'NAI response timeout'}