gokaygokay commited on
Commit
5af1f4a
·
verified ·
1 Parent(s): fa09fe7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import tempfile
3
  import time
@@ -92,6 +93,7 @@ def generate_3d_model(input_image):
92
  trimesh_mesh.export(tmp_file.name, file_type="glb", include_normals=True)
93
  return tmp_file.name
94
 
 
95
  def process_and_generate(prompt, height, width, steps, scales, seed):
96
  # Generate image from prompt
97
  generated_image = generate_image(prompt, height, width, steps, scales, seed)
 
1
+ import spaces
2
  import os
3
  import tempfile
4
  import time
 
93
  trimesh_mesh.export(tmp_file.name, file_type="glb", include_normals=True)
94
  return tmp_file.name
95
 
96
+ @spaces.GPU
97
  def process_and_generate(prompt, height, width, steps, scales, seed):
98
  # Generate image from prompt
99
  generated_image = generate_image(prompt, height, width, steps, scales, seed)