cavargas10 commited on
Commit
df3e7f6
·
verified ·
1 Parent(s): e0ccbf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -67,6 +67,7 @@ def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
67
  def get_seed(randomize_seed: bool, seed: int) -> int:
68
  return np.random.randint(0, MAX_SEED) if randomize_seed else seed
69
 
 
70
  def text_to_3d(
71
  prompt: str,
72
  seed: int,
@@ -99,6 +100,7 @@ def text_to_3d(
99
  torch.cuda.empty_cache()
100
  return state, video_path
101
 
 
102
  def extract_glb(
103
  state: dict,
104
  mesh_simplify: float,
 
67
  def get_seed(randomize_seed: bool, seed: int) -> int:
68
  return np.random.randint(0, MAX_SEED) if randomize_seed else seed
69
 
70
+ @spaces.GPU
71
  def text_to_3d(
72
  prompt: str,
73
  seed: int,
 
100
  torch.cuda.empty_cache()
101
  return state, video_path
102
 
103
+ @spaces.GPU
104
  def extract_glb(
105
  state: dict,
106
  mesh_simplify: float,