/w n models

app.py

@@ -1,15 +1,18 @@
 import os
 import gradio as gr
 from glob import glob
-
-cats = glob("models/coco_SDXL_simpleprompt/*")
+models = glob("models/*")
+#cats = glob("models/coco_SDXL_simpleprompt/*")
 
 
 with gr.Blocks() as demo:
-    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=[3], rows=[1], object_fit="contain", height="auto")
-        
+    for model in models:
+        with gr.Tab(model.split("/")[-1]) as tab:
+            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=[3], rows=[1], object_fit="contain", height="auto")
+