fix aspect ratio

app.py

@@ -13,14 +13,17 @@ from transformers import pipeline
 
 
 @interactive(
-    background_color=("green", ["green", "blue", "red"]),
-    border_size=(0.05, [0., 0.3]),
+    background_color=("green", ["green", "blue", "red"]),  # dropdown menu (str)
+    border_size=(0.05, [0., 0.3]),  # continuous slider (float)
 )
 def generate_background(background_color="green", border_size: float = 0.) -> np.ndarray:
-    out = np.zeros((256, 128, 3))  # Initial background set to black
+    out = np.zeros((256, 256, 3))  # Initial background set to black
     border_int = int(border_size * 256)
-    out[border_int:out.shape[0]-border_int, border_int:out.shape[1]-border_int,
-        ["red", "green", "blue"].index(background_color)] = 0.5
+    out[
+        border_int:out.shape[0]-border_int,
+        border_int:out.shape[1]-border_int,
+        ["red", "green", "blue"].index(background_color)
+    ] = 0.5
     return out