ZhiyuanthePony commited on
Commit
bab9f24
·
1 Parent(s): 41f5c58
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -56,6 +56,7 @@ def initialize_pipeline():
56
  print("Pipeline initialized!")
57
  return PIPELINE
58
 
 
59
  def generate_3d_mesh(prompt: str) -> Tuple[str, str]:
60
  """Generate 3D mesh from text prompt"""
61
  global PIPELINE, OBJ_FILE_QUEUE
 
56
  print("Pipeline initialized!")
57
  return PIPELINE
58
 
59
+ @spaces.GPU
60
  def generate_3d_mesh(prompt: str) -> Tuple[str, str]:
61
  """Generate 3D mesh from text prompt"""
62
  global PIPELINE, OBJ_FILE_QUEUE