sagemaker-launcher / utils /load_models.py
philschmid's picture
philschmid HF staff
online trainer
25f0c96
import requests
import streamlit as st
@st.cache
def load_models():
res = requests.get("https://huggingface.co/api/models").json()
return [model["modelId"] for model in res]