SeeMODEL / README.md
priyadip's picture
Add app.py and fix short_description YAML metadata
297ccd3

A newer version of the Gradio SDK is available: 6.12.0

Upgrade
metadata
title: Neural Model Analyzer
emoji: πŸ”¬
colorFrom: blue
colorTo: purple
sdk: gradio
sdk_version: 6.9.0
app_file: app.py
pinned: true
license: mit
short_description: 'Advanced model introspection: 150+ archs, weights'
tags:
  - model-analysis
  - pytorch
  - visualization
  - deep-learning

\U0001f52c Neural Model Analyzer

Upload any model file and get a deep analysis β€” architecture detection (150+ families), weight distributions, layer connectivity, memory profiling, and more. Runs on free HuggingFace CPU Spaces.

Supported Formats

Format Extension Notes
PyTorch .pth, .pt State dicts, full models, checkpoints
HuggingFace .bin Standard HF model format
SafeTensors .safetensors Fast, safe tensor format
ONNX .onnx With input/output specs

150+ Detected Architecture Families

NLP Encoders: BERT, RoBERTa, DistilBERT, ALBERT, ELECTRA, DeBERTa, XLM-R, Longformer, BigBird, MPNet, ConvBERT, MobileBERT, ...

Sentence Embeddings: Sentence-BERT, E5, BGE, GTE, Instructor, Jina, ...

Causal LM / Decoders: GPT-2, GPT-NeoX, GPT-J, LLaMA 1/2/3, Mistral, Mixtral (MoE), Phi, Qwen, Falcon, BLOOM, OPT, Mamba, RWKV, ChatGLM, DeepSeek, Yi, Baichuan, InternLM, StableLM, DBRX, OLMo, CodeLlama, StarCoder, ...

Seq2Seq: T5, Flan-T5, BART, mBART, Pegasus, MarianMT, NLLB, CodeT5, LED, ...

Vision: ViT, DeiT, BEiT, Swin, ConvNeXt, ResNet, EfficientNet, MobileNet, DINOv2, EVA, PVT, MaxViT, MLP-Mixer, ...

Detection & Segmentation: DETR, RT-DETR, YOLO, Faster R-CNN, SAM, SAM 2, SegFormer, Mask2Former, DeepLab, ...

Audio: Whisper, Wav2Vec2, HuBERT, WavLM, SpeechT5, Bark, VITS, MusicGen, AudioLDM, CLAP, ...

Multimodal: CLIP, OpenCLIP, SigLIP, BLIP, BLIP-2, LLaVA, InternVL, PaliGemma, CogVLM, Florence, Kosmos, ...

Generative: Stable Diffusion, SDXL, SD3, FLUX, DiT, PixArt, ControlNet, IP-Adapter, VQGAN, StyleGAN, ESRGAN, ...

Science: ESM (protein), AlphaFold-style, GNN, SchNet, Graphormer, MolBART, ...

PEFT: LoRA, QLoRA, IA3, Prefix Tuning, Prompt Tuning, Adapters, ...

And more: Time series (PatchTST, Informer), Video (VideoMAE, CogVideo), Document (Donut, TrOCR), RL (Decision Transformer), Quantized (GPTQ, AWQ, BnB), ...

Analysis Tabs

  1. Summary & Architecture β€” File info, param count, auto-detected architecture with confidence, inferred config (hidden size, layers, heads, vocab, etc.)
  2. Layer Tree β€” Full hierarchical view with shapes and sizes
  3. Types & Connections β€” Layer type inference + connectivity (direct, reshape, skip/residual)
  4. Weight Statistics β€” Per-tensor mean, std, sparsity, health warnings
  5. Distributions β€” Histograms per layer
  6. Module Sizes β€” Bar chart by module
  7. Heatmap β€” Normalized statistics across layers
  8. Memory β€” Dtype breakdown + largest tensors
  9. Depth Profile β€” Parameter count across network depth