Update app.py

app.py

@@ -252,7 +252,7 @@ if choose == "Collection2Collection":
         with st.spinner(text=f"Generating input images..."):
             punks = model.generate_app(
                 num=timestamped_filename(),
-                nrow=4,
+                nrow=nrows,
                 checkpoint=-1,
                 types="default"
             )[1]