NicolasG2523 commited on
Commit
15de5ac
·
verified ·
1 Parent(s): 4cfbead

Upload 2 files

Browse files
Files changed (2) hide show
  1. README.md +2 -1
  2. gradio_app.py +5 -6
README.md CHANGED
@@ -4,7 +4,8 @@ emoji: 🌍
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
- sdk_version: 4.44.1
 
8
  app_file: gradio_app.py
9
  pinned: false
10
  short_description: MultiImages-to-3D Generation
 
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
+ #sdk_version: 4.44.1
8
+ sdk_version: 5.33.1
9
  app_file: gradio_app.py
10
  pinned: false
11
  short_description: MultiImages-to-3D Generation
gradio_app.py CHANGED
@@ -33,7 +33,6 @@ MAX_SEED = 1e7
33
 
34
 
35
  import spaces
36
- from spaces import GPU
37
  import subprocess
38
  import sys
39
  import shlex
@@ -43,7 +42,7 @@ print('install custom')
43
  subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
44
 
45
 
46
- @GPU
47
  def dummy_gpu_function():
48
  pass
49
 
@@ -146,8 +145,7 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
146
  </div>
147
  """
148
 
149
- #@spaces.GPU(duration=40)
150
- @GPU(duration=40)
151
  def _gen_shape(
152
  caption=None,
153
  image=None,
@@ -254,7 +252,7 @@ def _gen_shape(
254
  main_image = image if not MV_MODE else image['front']
255
  return mesh, main_image, save_folder, stats, seed
256
 
257
- @GPU(duration=90)
258
  def generation_all(
259
  caption=None,
260
  image=None,
@@ -319,7 +317,7 @@ def generation_all(
319
  seed,
320
  )
321
 
322
- @GPU(duration=40)
323
  def shape_generation(
324
  caption=None,
325
  image=None,
@@ -366,6 +364,7 @@ def shape_generation(
366
  )
367
 
368
 
 
369
  def build_app():
370
  title = 'Hunyuan3D-2: High Resolution Textured 3D Assets Generation'
371
  if MV_MODE:
 
33
 
34
 
35
  import spaces
 
36
  import subprocess
37
  import sys
38
  import shlex
 
42
  subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
43
 
44
 
45
+ @spaces.GPU
46
  def dummy_gpu_function():
47
  pass
48
 
 
145
  </div>
146
  """
147
 
148
+ @spaces.GPU(duration=40)
 
149
  def _gen_shape(
150
  caption=None,
151
  image=None,
 
252
  main_image = image if not MV_MODE else image['front']
253
  return mesh, main_image, save_folder, stats, seed
254
 
255
+ @spaces.GPU(duration=90)
256
  def generation_all(
257
  caption=None,
258
  image=None,
 
317
  seed,
318
  )
319
 
320
+ @spaces.GPU(duration=40)
321
  def shape_generation(
322
  caption=None,
323
  image=None,
 
364
  )
365
 
366
 
367
+ @spaces.GPU
368
  def build_app():
369
  title = 'Hunyuan3D-2: High Resolution Textured 3D Assets Generation'
370
  if MV_MODE: