asserts for height and width

2023-01-11 14:00:01 +01:00 · 2023-01-11 14:00:01 +01:00 · 608f0f30a4
parent e50f713479
commit 608f0f30a4
1 changed files with 3 additions and 3 deletions
--- a/gradio_ui.py
+++ b/gradio_ui.py
@ -187,9 +187,9 @@ class BlendingFrontend():
            list_imgs = [(255*np.random.rand(self.height,self.width,3)).astype(np.uint8) for l in range(5)]
            self.imgs_show_current = copy.deepcopy(list_imgs)
            return list_imgs
-        # FIXME TODO ASSERTS
-        self.lb.sdh.height = self.height
-        self.lb.sdh.width = self.width
+        
+        self.lb.set_width(self.width)
+        self.lb.set_height(self.height)
        
        self.lb.autosetup_branching(
                depth_strength = self.depth_strength,