update

app.py

@@ -27,8 +27,8 @@ with gr.Blocks(css="style.css") as demo:
         visible=SHOW_DUPLICATE_BUTTON,
     )
     with gr.Tabs():
-        # with gr.TabItem("Depth"):
-        #     create_demo_depth(model.process_depth)
+        with gr.TabItem("Depth"):
+            create_demo_depth(model.process_depth)
         with gr.TabItem("Canny"):
             create_demo_canny(model.process_canny)
 

model.py

@@ -114,6 +114,9 @@ class Model:
         image = resize_image_to_16_multiple(image, 'canny')
         W, H = image.size
         print(W, H)
+        self.t5_model.model.to(self.device)
+        self.gpt_model_canny.to(self.device)
+
         condition_img = self.get_control_canny(np.array(image), low_threshold,
                                                high_threshold)
         condition_img = torch.from_numpy(condition_img[None, None,