Spaces:
Sleeping
Sleeping
Commit
·
f6a3903
1
Parent(s):
85dc93e
update
Browse files
model.py
CHANGED
@@ -28,6 +28,7 @@ import subprocess
|
|
28 |
from rgb2x.pipeline_rgb2x import StableDiffusionAOVMatEstPipeline
|
29 |
from app_texnet import image_to_temp_path
|
30 |
import os
|
|
|
31 |
|
32 |
CONTROLNET_MODEL_IDS = {
|
33 |
# "Openpose": "lllyasviel/control_v11p_sd15_openpose",
|
@@ -317,7 +318,9 @@ class Model:
|
|
317 |
normal_map_path = image_to_temp_path(preds[0].rotate(90), "normal_map")
|
318 |
roughness_path = image_to_temp_path(preds[1].rotate(90), "roughness")
|
319 |
metallic_path = image_to_temp_path(preds[2].rotate(90), "metallic")
|
320 |
-
|
|
|
|
|
321 |
os.makedirs(os.path.dirname(output_blend_path), exist_ok=True)
|
322 |
|
323 |
def run_blend_generation(
|
|
|
28 |
from rgb2x.pipeline_rgb2x import StableDiffusionAOVMatEstPipeline
|
29 |
from app_texnet import image_to_temp_path
|
30 |
import os
|
31 |
+
import time
|
32 |
|
33 |
CONTROLNET_MODEL_IDS = {
|
34 |
# "Openpose": "lllyasviel/control_v11p_sd15_openpose",
|
|
|
318 |
normal_map_path = image_to_temp_path(preds[0].rotate(90), "normal_map")
|
319 |
roughness_path = image_to_temp_path(preds[1].rotate(90), "roughness")
|
320 |
metallic_path = image_to_temp_path(preds[2].rotate(90), "metallic")
|
321 |
+
prompt_nospace = prompt.replace(' ', '_')
|
322 |
+
current_timecode = time.strftime("%Y%m%d_%H%M%S")
|
323 |
+
output_blend_path = os.path.join(os.getcwd(), "output", f"{obj_name}_{prompt_nospace}_{current_timecode}.blend") # replace with desired output path
|
324 |
os.makedirs(os.path.dirname(output_blend_path), exist_ok=True)
|
325 |
|
326 |
def run_blend_generation(
|