restart

app.py

@@ -11,7 +11,13 @@ with gr.Blocks() as demo:
             cats = glob(model+"/*")
             for cat in cats:
                 with gr.Tab(cat.split("/")[-1]) as tab:
-                    gallery = gr.Gallery(value=glob(os.path.join(cat, "*.jpg")),label="Generated images", show_label=False, elem_id="gallery",columns=[5], rows=[1], object_fit="contain", height="auto")
+                    gallery = gr.Gallery(value=glob(os.path.join(cat, "*.jpg")),
+                                         label="Generated images",
+                                         show_label=False, elem_id="gallery",
+                                         columns=[5],
+                                         rows=[1],
+                                         object_fit="contain",
+                                         height="auto")