Spaces:
Running
Running
Commit
·
7511f27
1
Parent(s):
9933c91
app.py: cached uploaded images.
Browse files
app.py
CHANGED
|
@@ -218,6 +218,21 @@ def load_example (face_file, pose_file, prompt, style, negative_prompt):
|
|
| 218 |
return image, gr.update(visible=True)
|
| 219 |
|
| 220 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 221 |
def generate_image (
|
| 222 |
novita_key1,
|
| 223 |
model_name,
|
|
@@ -274,13 +289,15 @@ def generate_image (
|
|
| 274 |
CONTROLNET_DICT['canny']['strength'] = canny_strength
|
| 275 |
CONTROLNET_DICT['depth']['strength'] = depth_strength
|
| 276 |
|
|
|
|
|
|
|
| 277 |
res = client._post('/v3/async/instant-id', {
|
| 278 |
'extra': {
|
| 279 |
'response_image_type': 'jpeg',
|
| 280 |
},
|
| 281 |
'model_name': f'{model_name}.safetensors',
|
| 282 |
-
'face_image_assets_ids':
|
| 283 |
-
'ref_image_assets_ids':
|
| 284 |
'prompt': prompt,
|
| 285 |
'negative_prompt': negative_prompt,
|
| 286 |
'controlnet': {
|
|
|
|
| 218 |
return image, gr.update(visible=True)
|
| 219 |
|
| 220 |
|
| 221 |
+
upload_depot = {}
|
| 222 |
+
|
| 223 |
+
def upload_assets_with_cache (client, paths):
|
| 224 |
+
global upload_depot
|
| 225 |
+
|
| 226 |
+
pending_paths = [path for path in paths if not path in upload_depot]
|
| 227 |
+
if pending_paths:
|
| 228 |
+
print('uploading images:', pending_paths)
|
| 229 |
+
for key, value in zip(pending_paths, client.upload_assets(pending_paths)):
|
| 230 |
+
upload_depot[key] = value
|
| 231 |
+
|
| 232 |
+
return [upload_depot[path] for path in paths]
|
| 233 |
+
|
| 234 |
+
|
| 235 |
+
|
| 236 |
def generate_image (
|
| 237 |
novita_key1,
|
| 238 |
model_name,
|
|
|
|
| 289 |
CONTROLNET_DICT['canny']['strength'] = canny_strength
|
| 290 |
CONTROLNET_DICT['depth']['strength'] = depth_strength
|
| 291 |
|
| 292 |
+
face_image_uploaded, ref_image_uploaded = upload_assets_with_cache(client, [face_image_path, ref_image_path])
|
| 293 |
+
|
| 294 |
res = client._post('/v3/async/instant-id', {
|
| 295 |
'extra': {
|
| 296 |
'response_image_type': 'jpeg',
|
| 297 |
},
|
| 298 |
'model_name': f'{model_name}.safetensors',
|
| 299 |
+
'face_image_assets_ids': [face_image_uploaded],
|
| 300 |
+
'ref_image_assets_ids': [ref_image_uploaded],
|
| 301 |
'prompt': prompt,
|
| 302 |
'negative_prompt': negative_prompt,
|
| 303 |
'controlnet': {
|