Skip to content

Tool to track AI models suitable for local use (i.e. under 24GB VRAM) on Hugging Space and Modelscope

License

Notifications You must be signed in to change notification settings

nigelp/ai-model-tracker

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

9 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ€– AI Model Tracker

Python Version License GitHub Release

aitrackerscreen

AI Model Tracker is a one-click desktop application that automatically discovers and organizes AI models from across the internet. It solves the problem of "model sprawl" where you have models scattered everywhere without knowing what you have, what's new, or what will work on your hardware.

✨ Key Features

  • Automatic Discovery - Scrapes Hugging Face, ModelScope (Chinese models), and more for new AI models
  • Hardware-Aware - Focuses on models that work with ≀24GB VRAM (most consumer GPUs)
  • Beautiful Dashboard - Web interface to browse, filter, and search models
  • Chinese Model Tracking - Special attention to Qwen, DeepSeek, and other Chinese models
  • Size Estimates - Rough GB estimates so you know download sizes
  • GGUF Metadata - Extracts detailed GGUF quantization information
  • Auto-Launch - Automatically opens browser when started
  • Weekly Digests - Automated reports of new models

πŸš€ Quick Start

Windows Users

An executable file is available in Releases. But for those who would rather do things the manual way -

  1. Clone the repository

    git clone https://github.com/nigelp/ai-model-tracker.git
    cd ai-model-tracker
  2. Install

    install.bat
  3. Start

    start_tracker.bat
  4. Open your browser to http://localhost:5000

Linux/Mac Users

  1. Clone the repository

    git clone https://github.com/nigelp/ai-model-tracker.git
    cd ai-model-tracker
  2. Install

    chmod +x install.sh
    ./install.sh
  3. Start

    chmod +x start_tracker.sh
    ./start_tracker.sh
  4. Open your browser to http://localhost:5000

πŸ“ Project Structure

ai-model-tracker/
β”œβ”€β”€ web_dashboard.py      # Web interface with Flask
β”œβ”€β”€ model_scraper.py      # Model discovery scraper
β”œβ”€β”€ gguf_parser.py       # GGUF metadata extraction
β”œβ”€β”€ install.bat          # Windows installer
β”œβ”€β”€ start_tracker.bat    # Windows launcher
β”œβ”€β”€ config.json          # Configuration settings
β”œβ”€β”€ requirements.txt     # Python dependencies
β”œβ”€β”€ web_dashboard.spec   # PyInstaller spec
β”œβ”€β”€ data/                # SQLite database (gitignored)
β”œβ”€β”€ tools/              # External binaries
└── reports/            # Generated reports (gitignored)

πŸ–₯️ Using the Dashboard

Once running, you'll see a professional dashboard with:

Left Panel - Statistics:

  • Total models tracked
  • Chinese vs. non-Chinese count
  • Breakdown by category (Text, Image, Video, Audio, Coding)
  • GGUF model count

Middle - Filter Controls:

  • Filter by source: Hugging Face, ModelScope
  • Filter by category
  • Show only Chinese models
  • Filter by GGUF format
  • Search by name or description
  • Sort by date, downloads, likes, name, or size

Right - Model Cards: Each model shows:

  • Name and brief description
  • Estimated size (helpful for download planning)
  • Release date
  • Source badge (HF, MS)
  • GGUF badge with quantization details
  • VRAM requirements
  • Context length
  • Chinese indicator if applicable
  • Direct link to model page

Top Right - Actions:

  • Refresh data button
  • Export to JSON option

βš™οΈ Configuration

Edit config.json to change settings:

{
  "scrape_interval_hours": 6,        // How often to refresh
  "max_models_per_source": 100,      // Models per source
  "vram_limit_gb": 24,             // VRAM limit for filtering
  "include_chinese": true,           // Include Chinese models
  "sources": {
    "huggingface": true,             // Enable HF scraping
    "modelscope": true               // Enable ModelScope scraping
  }
}

πŸ› οΈ Building the Executable

If you want to build the standalone Windows executable:

# Install PyInstaller
pip install pyinstaller

# Build
pyinstaller web_dashboard.spec --clean

# Result: dist/web_dashboard.exe (includes auto-launch browser)

πŸ› Common Problems

"Python not found" Error

Problem: Installer can't find Python Solution: Install Python 3.8+ from python.org

  • Windows: Check "Add Python to PATH" during installation

"Port 5000 already in use"

Problem: Another app is using port 5000 Solution: Change port in web_dashboard.py line 619:

app.run(debug=True, port=5001)  # Change to 5001

Then use http://localhost:5001 in your browser

"Permission denied" on Mac/Linux

Problem: Scripts aren't executable Solution: Run this command:

chmod +x install.sh start_tracker.sh

"Module not found" (Flask/Requests)

Problem: Python packages missing Solution: Install manually:

pip install -r requirements.txt

Or re-run the installer

"Connection error" or "No models showing"

Problem: Internet issue or API limits Solution:

  • Check your internet connection
  • Wait a minute and refresh
  • Some sources have rate limits

πŸ“Š Initial Data

The system comes with sample data including:

  • Real Hugging Face trending models (live data)
  • Chinese models: Qwen2.5, DeepSeek-Coder, etc.
  • Image models: Stable Diffusion 3 example
  • Multimodal models: Llama 3.2 Vision example
  • Size estimates for each model

πŸ”„ Updates

Manual Update: Click "Refresh" in dashboard

Automatic Update: The scraper runs every 6 hours automatically

Force Update: Stop the app (Ctrl+C) and restart it

πŸ“§ Reports

The system automatically generates a weekly HTML report showing:

  • New models from past 7 days
  • Statistics and trends
  • Chinese model highlights
  • Size distribution

Find it at: http://localhost:5000/weekly-report or in the reports/ folder.

πŸ—‚οΈ Data Storage

  • Database: data/models.db (SQLite file)
  • Reports: reports/weekly_report.html
  • Configuration: config.json

You can safely delete the data folder to start fresh.

πŸ‘₯ Ideal For

  • AI enthusiasts who want to stay updated on new models
  • Developers looking for specific model types
  • Researchers tracking model release trends
  • Anyone tired of manually checking multiple websites

πŸ’‘ Pro Tips

  • Use search to find specific model types
  • Filter by "Chinese Only" to see Qwen/DeepSeek models
  • Check size estimates before downloading huge models
  • Export data if you want to analyze in a spreadsheet
  • The system works offline once data is collected

🀝 Acknowledgments

This project uses the following open-source tools:

  • gpustack/gguf-parser-go - GGUF metadata extraction
    • This project would not have detailed GGUF support without this excellent parser
    • Licensed under Apache 2.0

πŸ“ License

This project is licensed under the MIT License - see the LICENSE file for details.

πŸ› Getting Help

  • Check the console window for error messages
  • Ensure Python 3.8+ is installed and in PATH
  • Try manual installation with pip install -r requirements.txt
  • Check firewall/antivirus isn't blocking Python

πŸ“ˆ Next Steps After Installation

Once you have it running, you can:

  • Bookmark http://localhost:5000 in your browser
  • Set up automatic startup (add to startup folder on Windows)
  • Share weekly reports with your team
  • Customize sources in config.json

🏁 Getting Started Checklist

  • Python 3.8+ installed
  • Repository cloned
  • Ran installer (install.bat or install.sh)
  • Started tracker (start_tracker.bat or start_tracker.sh)
  • Browser opened to http://localhost:5000
  • Can see model cards and filters
  • Refresh data successfully

Enjoy tracking AI models! πŸš€

About

Tool to track AI models suitable for local use (i.e. under 24GB VRAM) on Hugging Space and Modelscope

Resources

License

Stars

Watchers

Forks

Packages

No packages published