update space

app.py

@@ -14,6 +14,7 @@ import random
 import warnings
 from huggingface_hub import hf_hub_download
 import hashlib
+import spaces
 
 import sys
 
@@ -65,6 +66,9 @@ def optimize_all(xm, models, initial_noise, noise_start_t, diffusion, latent_mod
     edited_1, edited_2, edited_3, edited_4, state = _3d_editing(xm, models, diffusion, initial_noise, noise_start_t, device, instruction, rand_seed, state)
     print(state)
     return out_gen_1, out_gen_2, out_gen_3, out_gen_4, edited_1, edited_2, edited_3, edited_4
+
+
+@spaces.GPU()
 @torch.no_grad()
 def generate_3d_with_shap_e(xm, diffusion, latent_model, device, prompt, rand_seed, state):
     set_seed(rand_seed)
@@ -119,6 +123,8 @@ def generate_3d_with_shap_e(xm, diffusion, latent_model, device, prompt, rand_se
         mesh_path.append(output_path_tmp)
     del latents
     return mesh_path[0], mesh_path[1], mesh_path[2], mesh_path[3], state
+
+@spaces.GPU()
 @torch.no_grad()
 def _3d_editing(xm, models, diffusion, initial_noise, start_t, device, instruction, rand_seed, state):
     set_seed(rand_seed)
@@ -203,6 +209,7 @@ def _3d_editing(xm, models, diffusion, initial_noise, start_t, device, instructi
             mesh.export(output_path)
             mesh_path.append(output_path)
     return mesh_path[0], mesh_path[1], mesh_path[2], mesh_path[3], state
+
 def main():
 
     css = """