Update README.md

README.md

@@ -9,6 +9,11 @@ tags:
 ---
 
 # Automodel Now Available
+
+
+![image](https://cdn-uploads.huggingface.co/production/uploads/630cf55b15433862cfc9556f/92PPQUxtHkYnyNQgF9Wtm.png)
+
+
 ```
 # ── Test AutoModel loading + generation ──
 from transformers import AutoModel
@@ -28,7 +33,7 @@ for c in range(10):
     imgs = model.sample(n_samples=4, class_label=c)
     all_imgs.append(imgs)
 
-grid = make_grid(torch.cat(all_imgs), nrow=10)
+grid = make_grid(torch.cat(all_imgs), nrow=4)
 save_image(grid, "automodel_test.png")
 print("✓ Saved automodel_test.png — 4 per class, 10 classes")
 ```