Spaces:
Running
Running
承弱
commited on
Commit
·
4b97602
1
Parent(s):
f6c784d
add timeout fix
Browse files- src/generation.py +5 -2
src/generation.py
CHANGED
@@ -38,8 +38,11 @@ def call_generation(prompt, mask_image_url,lora_path_ratio="0 1.0", image_width=
|
|
38 |
all_res_ = []
|
39 |
REPEAT = 1
|
40 |
for _ in range(REPEAT):
|
41 |
-
|
42 |
-
|
|
|
|
|
|
|
43 |
|
44 |
all_image_data = []
|
45 |
for res_ in all_res_:
|
|
|
38 |
all_res_ = []
|
39 |
REPEAT = 1
|
40 |
for _ in range(REPEAT):
|
41 |
+
try:
|
42 |
+
res_ = requests.post(url_create_task, data=json.dumps(data), headers=headers, timeout=60)
|
43 |
+
all_res_.append(res_)
|
44 |
+
except:
|
45 |
+
pass
|
46 |
|
47 |
all_image_data = []
|
48 |
for res_ in all_res_:
|