Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|