From 608f0f30a496a986fb632dc1e246880f6a670231 Mon Sep 17 00:00:00 2001
From: Johannes Stelzer <stelzer@lunar-ring.ai>
Date: Wed, 11 Jan 2023 14:00:01 +0100
Subject: [PATCH] asserts for height and width

---
 gradio_ui.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/gradio_ui.py b/gradio_ui.py
index f003c60..a723332 100644
--- 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,