Update app.py

app.py

@@ -166,8 +166,14 @@ window.addEventListener('load', () => {
 </script>
 """
 
+CSS="""
+<style>
+    .gallery { min_width=512px; min_height=512px; max_height=512px; !important; }
+</style>
+"""
+
 # with gr.Blocks(fill_width=True, head=js) as demo:
-with gr.Blocks(theme="JohnSmith9982/small_and_pretty") as demo:
+with gr.Blocks(css=CSS) as demo:
     with gr.Tab(str(num_models) + ' Models'):
         with gr.Column(scale=2):
             with gr.Group():