Update model.py

model.py

@@ -119,8 +119,8 @@ 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)
+        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)