alexnasa commited on
Commit
ac4feba
·
verified ·
1 Parent(s): b3f7f27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -12
app.py CHANGED
@@ -65,7 +65,7 @@ def sh(cmd_list, extra_env=None):
65
 
66
  # install with FORCE_CUDA=1
67
  sh(["pip", "install", "diso"], {"FORCE_CUDA": "1"})
68
- sh(["pip", "install", "torch-cluster", "-f", "https://data.pyg.org/whl/torch-2.7.0+126.html"])
69
 
70
 
71
 
@@ -154,17 +154,6 @@ def run_triposg(image_path: str,
154
  merged.export(glb_path)
155
 
156
  mesh_file = first_file_from_dir(export_dir, "glb")
157
-
158
- print(mesh_file)
159
-
160
- # 1) Check for the file’s existence
161
- if not os.path.exists(glb_path):
162
- raise FileNotFoundError(f"No merged .glb found at {glb_path}")
163
-
164
- # 2) List every file in the folder
165
- all_files = os.listdir(export_dir)
166
-
167
- print(f"Files in {export_dir}: {all_files}")
168
 
169
  return mesh_file, export_dir
170
 
@@ -203,9 +192,31 @@ def build_demo():
203
  remove_bg = gr.Checkbox(value=False, label="Remove Background (RMBG)")
204
 
205
  with gr.Column(scale=1):
 
 
 
 
 
 
 
206
  output_model = gr.Model3D(label="Merged 3D Object")
207
  output_dir = gr.Textbox(label="Export Directory")
208
  examples = gr.Examples(
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
209
  examples=[
210
  [
211
  "assets/images/np4_7bd5d25aa77b4fb18e780d7a4c97d342.png",
 
65
 
66
  # install with FORCE_CUDA=1
67
  sh(["pip", "install", "diso"], {"FORCE_CUDA": "1"})
68
+ # sh(["pip", "install", "torch-cluster", "-f", "https://data.pyg.org/whl/torch-2.7.0+126.html"])
69
 
70
 
71
 
 
154
  merged.export(glb_path)
155
 
156
  mesh_file = first_file_from_dir(export_dir, "glb")
 
 
 
 
 
 
 
 
 
 
 
157
 
158
  return mesh_file, export_dir
159
 
 
192
  remove_bg = gr.Checkbox(value=False, label="Remove Background (RMBG)")
193
 
194
  with gr.Column(scale=1):
195
+ gr.HTML(
196
+ """
197
+ <p style="opacity: 0.6; font-style: italic;">
198
+ This might take a few seconds to load the 3D model
199
+ </p>
200
+ """
201
+ )
202
  output_model = gr.Model3D(label="Merged 3D Object")
203
  output_dir = gr.Textbox(label="Export Directory")
204
  examples = gr.Examples(
205
+
206
+ examples=[
207
+ [
208
+ "assets/images/np5_b81f29e567ea4db48014f89c9079e403.png",
209
+ 10,
210
+ ],
211
+
212
+ ],
213
+ examples=[
214
+ [
215
+ "assets/images/np10_cc486e491a2c499f9fd2aad2b02c6ccb.png",
216
+ 10,
217
+ ],
218
+
219
+ ],
220
  examples=[
221
  [
222
  "assets/images/np4_7bd5d25aa77b4fb18e780d7a4c97d342.png",