search

google-ai-edge / model-explorer

A modern model graph visualizer and debugger
Apache License 2.0
961 stars 68 forks source link

Hugging Face Spaces integration #102

Open osanseviero opened 1 month ago

osanseviero commented 1 month ago

Hello :wave: It would be cool to explore some kind of integration with Hugging Face Spaces :hugs:

Recently, a community member built an example of the model explorer hosted with a Space (https://huggingface.co/spaces/1aurent/model-explorer). It would be great to have an official Google template in Spaces.

Additionally, it would be very nice if one could select TF/TFlite/TFJSON/etc shared on the HF Hub (e.g. https://huggingface.co/models?library=jax&sort=trending for JAX or https://huggingface.co/models?library=tflite&sort=trending for TF Lite) to get started with the model explorer very quickly. For example, I downloaded DistilBERT tflite and uploaded it to the Space. Maybe we can skip the download and upload step entirely :exploding_head:

Screenshot from 2024-07-12 16-09-43

pkgoogle commented 1 month ago

That's very cool! Let me discuss it with the team on how to best utilize this.