Update app.py

app.py

@@ -145,9 +145,11 @@ def preprocess_image(image: Image.Image,
     new_width, new_height = int(width * ratio), int(height * ratio)
     image = image['composite'].resize((new_width, new_height))
 
+    print("image:",type(image))
+
     prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
     
-    print(prompt, negative_prompt, style_name, num_steps, guidance_scale, controlnet_conditioning_scale)
+    print("params:", prompt, negative_prompt, style_name, num_steps, guidance_scale, controlnet_conditioning_scale)
     image = pipe_control(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -535,6 +537,8 @@ with gr.Blocks(delete_cache=(600, 600), js=js_func) as demo:
 if __name__ == "__main__":
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
     pipeline.cuda()
+
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     
     #scribble controlnet
     controlnet = ControlNetModel.from_pretrained(
@@ -551,7 +555,7 @@ if __name__ == "__main__":
         # scheduler=eulera_scheduler,
     )
     pipe_control.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_control.scheduler.config)
-    pipe_control.to("cuda")
+    pipe_control.to(device)
     
     # try:
     #     pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg