Update app.py

app.py

@@ -98,14 +98,15 @@ infer_config = config.infer_config
 
 IS_FLEXICUBES = True if config_name.startswith('instant-mesh') else False
 
-device = torch.device('cuda')
+# Set device to CPU
+device = torch.device('cpu')
 
 # load diffusion model
 print('Loading diffusion model ...')
 pipeline = DiffusionPipeline.from_pretrained(
     "sudo-ai/zero123plus-v1.2", 
     custom_pipeline="zero123plus",
-    torch_dtype=torch.float16,
+    torch_dtype=torch.float32,  # Changed from float16 to float32 for CPU
 )
 pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
     pipeline.scheduler.config, timestep_spacing='trailing'
@@ -137,7 +138,6 @@ def check_input_image(input_image):
 
 
 def preprocess(input_image, do_remove_background):
-
     rembg_session = rembg.new_session() if do_remove_background else None
 
     if do_remove_background:
@@ -147,9 +147,7 @@ def preprocess(input_image, do_remove_background):
     return input_image
 
 
-@spaces.GPU
 def generate_mvs(input_image, sample_steps, sample_seed):
-
     seed_everything(sample_seed)
     
     # sampling
@@ -167,9 +165,7 @@ def generate_mvs(input_image, sample_steps, sample_seed):
     return z123_image, show_image
 
 
-@spaces.GPU
 def make3d(images):
-
     global model
     if IS_FLEXICUBES:
         model.init_flexicubes_geometry(device, use_renderer=False)