Update gen_demo.py

gen_demo.py

@@ -147,7 +147,7 @@ with gr.Blocks() as demo:
             
             # Dropdown for class selection
             class_dropdown = gr.Dropdown(
-                choices=class_choices[:20],  # Limit for demonstration
+                choices=class_choices,  # Limit for demonstration
                 label="Select ImageNet Class",
                 value=class_choices[0] if class_choices else None
             )