Update app.py

app.py

@@ -1,11 +1,17 @@
 import gradio as gr
 from random import randint
-from all_models import model_groups
+from all_models import models
 from datetime import datetime
 
 kii = "mohawk femboy racecar driver"
 
 # Predefined arrays of models
+model_groups = {
+    "Group A": models[:3],  # Replace with specific subsets
+    "Group B": models[3:6],
+    "Group C": models[6:9],
+    "Group D": models[9:12],
+}
 
 
 def get_current_time():
@@ -119,7 +125,7 @@ js_code = """
     };
     setTimeout(() => {
         window.scrollTo = originalScroll;
-    }, 3); // Restore scroll function after 3 seconds
+    }, 3000); // Restore scroll function after 3 seconds
 </script>
 """