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(
|