Update app.py

app.py

@@ -24,13 +24,13 @@ from huggingface_hub import hf_hub_download
 # pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
 # model = model.cuda()
 # ddim_sampler = DDIMSampler(model)
-model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
+model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml')
 ckpt = hf_hub_download(repo_id="xywwww/scene_diffusion", filename="checkpoints/epoch=25-step=112553.ckpt")
 model.load_state_dict(load_state_dict(ckpt), strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)
 
-
+@spaces.GPU
 def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):
     with torch.no_grad():
         img = resize_image(HWC3(input_image), image_resolution)