Spaces:
Sleeping
Sleeping
Commit
·
4f00e0c
1
Parent(s):
a643d17
Commit
Browse files
python.py
CHANGED
|
@@ -11,6 +11,7 @@ from custom_nodes.ComfyUI_Comfyroll_CustomNodes.node_mappings import NODE_CLASS_
|
|
| 11 |
from custom_nodes.ComfyUI_Comfyroll_CustomNodes.node_mappings import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_4
|
| 12 |
from comfy_extras.nodes_model_advanced import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_5
|
| 13 |
from comfy_extras.nodes_flux import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_6
|
|
|
|
| 14 |
|
| 15 |
from huggingface_hub import hf_hub_download
|
| 16 |
token = os.getenv("HUGGINGFACE_TOKEN")
|
|
@@ -265,5 +266,5 @@ def generate_image(prompt,
|
|
| 265 |
saveimage_9 = saveimage.save_images(
|
| 266 |
filename_prefix="image", images=get_value_at_index(vaedecode_8, 0)
|
| 267 |
)
|
| 268 |
-
saved_path = f"output/{saveimage_9['ui']['images'][0]['filename']}"
|
| 269 |
return saved_path, seed
|
|
|
|
| 11 |
from custom_nodes.ComfyUI_Comfyroll_CustomNodes.node_mappings import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_4
|
| 12 |
from comfy_extras.nodes_model_advanced import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_5
|
| 13 |
from comfy_extras.nodes_flux import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_6
|
| 14 |
+
import time
|
| 15 |
|
| 16 |
from huggingface_hub import hf_hub_download
|
| 17 |
token = os.getenv("HUGGINGFACE_TOKEN")
|
|
|
|
| 266 |
saveimage_9 = saveimage.save_images(
|
| 267 |
filename_prefix="image", images=get_value_at_index(vaedecode_8, 0)
|
| 268 |
)
|
| 269 |
+
saved_path = f"output/image_{int(time.time() * 1000)}_{saveimage_9['ui']['images'][0]['filename']}"
|
| 270 |
return saved_path, seed
|