staswrs commited on
Commit
f2228f5
·
1 Parent(s): 016e264

clean scene 3

Browse files
Files changed (2) hide show
  1. app.py +6 -4
  2. requirements.txt +1 -1
app.py CHANGED
@@ -21,7 +21,9 @@ import zipfile
21
  import requests
22
  import traceback
23
  import trimesh
 
24
  from trimesh.exchange.gltf import export_glb
 
25
  from inference_triposg import run_triposg
26
  from triposg.pipelines.pipeline_triposg import TripoSGPipeline
27
  from briarmbg import BriaRMBG
@@ -123,16 +125,16 @@ def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25):
123
  mesh.metadata.clear()
124
  mesh.name = "geometry_0"
125
 
 
 
 
 
126
  # Экспорт .glb вручную (иначе Trimesh добавляет сцену)
127
  # glb_data = mesh.export(file_type="glb")
128
  # with open(output_path, "wb") as f:
129
  # f.write(glb_data)
130
 
131
 
132
- glb_data = export_glb(mesh, include_scene=False)
133
- with open(output_path, "wb") as f:
134
- f.write(glb_data)
135
-
136
  print(f"[DEBUG] Mesh saved to {output_path}")
137
  return output_path if os.path.exists(output_path) else None
138
  # except Exception as e:
 
21
  import requests
22
  import traceback
23
  import trimesh
24
+ print("Trimesh version:", trimesh.__version__)
25
  from trimesh.exchange.gltf import export_glb
26
+
27
  from inference_triposg import run_triposg
28
  from triposg.pipelines.pipeline_triposg import TripoSGPipeline
29
  from briarmbg import BriaRMBG
 
125
  mesh.metadata.clear()
126
  mesh.name = "geometry_0"
127
 
128
+ glb_data = export_glb(mesh, include_scene=False)
129
+ with open(output_path, "wb") as f:
130
+ f.write(glb_data)
131
+
132
  # Экспорт .glb вручную (иначе Trimesh добавляет сцену)
133
  # glb_data = mesh.export(file_type="glb")
134
  # with open(output_path, "wb") as f:
135
  # f.write(glb_data)
136
 
137
 
 
 
 
 
138
  print(f"[DEBUG] Mesh saved to {output_path}")
139
  return output_path if os.path.exists(output_path) else None
140
  # except Exception as e:
requirements.txt CHANGED
@@ -2,7 +2,7 @@ gradio
2
  torch
3
  torchvision
4
  torchaudio
5
- trimesh
6
  pillow
7
  omegaconf
8
  diffusers
 
2
  torch
3
  torchvision
4
  torchaudio
5
+ trimesh==4.1.3
6
  pillow
7
  omegaconf
8
  diffusers