staswrs commited on
Commit
cec312b
·
1 Parent(s): e28dbf7

add octree depth controls fix

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -68,8 +68,8 @@ rmbg_net.eval()
68
 
69
 
70
  # def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25):
71
- def generate(image_path, face_number=50000, octree_depth=9, guidance_scale=5.0, num_steps=25):
72
- print(f"[INPUT] octree_depth={octree_depth}, face_number={face_number}, guidance_scale={guidance_scale}, num_steps={num_steps}")# 👈 добавлено
73
  print("[API CALL] image_path received:", image_path)
74
  print("[API CALL] File exists:", os.path.exists(image_path))
75
 
@@ -132,9 +132,9 @@ demo = gr.Interface(
132
  inputs=[
133
  gr.Image(type="filepath", label="Upload image"),
134
  gr.Slider(10000, 150000, step=10000, value=50000, label="Face count"),
135
- gr.Slider(6, 10, step=1, value=9, label="Octree Depth"),
136
  gr.Slider(1.0, 10.0, step=0.5, value=5.0, label="Guidance Scale"),
137
  gr.Slider(10, 100, step=5, value=25, label="Steps"),
 
138
  ], # 👈 добавлено
139
  outputs=gr.File(label="Download .glb"),
140
  title="TripoSG Image to 3D",
 
68
 
69
 
70
  # def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25):
71
+ def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25, octree_depth=9):
72
+ print(f"[INPUT] face_number={face_number}, guidance_scale={guidance_scale}, num_steps={num_steps}, octree_depth={octree_depth}")# 👈 добавлено
73
  print("[API CALL] image_path received:", image_path)
74
  print("[API CALL] File exists:", os.path.exists(image_path))
75
 
 
132
  inputs=[
133
  gr.Image(type="filepath", label="Upload image"),
134
  gr.Slider(10000, 150000, step=10000, value=50000, label="Face count"),
 
135
  gr.Slider(1.0, 10.0, step=0.5, value=5.0, label="Guidance Scale"),
136
  gr.Slider(10, 100, step=5, value=25, label="Steps"),
137
+ gr.Slider(6, 10, step=1, value=9, label="Octree Depth"),
138
  ], # 👈 добавлено
139
  outputs=gr.File(label="Download .glb"),
140
  title="TripoSG Image to 3D",