|
import streamlit as st |
|
from persist import persist, load_widget_state |
|
import pandas as pd |
|
import requests |
|
|
|
|
|
|
|
|
|
@st.cache |
|
def get_cached_data(): |
|
languages_df = pd.read_html("https://hf.co/languages")[0] |
|
languages_map = pd.Series(languages_df["Language"].values, index=languages_df["ISO code"]).to_dict() |
|
|
|
license_df = pd.read_html("https://huggingface.co/docs/hub/repositories-licenses")[0] |
|
license_map = pd.Series( |
|
license_df["License identifier (to use in model card)"].values, index=license_df.Fullname |
|
).to_dict() |
|
|
|
available_metrics = [x['id'] for x in requests.get('https://huggingface.co/api/metrics').json()] |
|
|
|
r = requests.get('https://huggingface.co/api/models-tags-by-type') |
|
tags_data = r.json() |
|
libraries = [x['id'] for x in tags_data['library']] |
|
tasks = [x['id'] for x in tags_data['pipeline_tag']] |
|
|
|
return license_map |
|
|
|
|
|
|
|
|
|
|
|
def main(): |
|
license_map= get_cached_data() |
|
|
|
st.markdown('## Model Details') |
|
st.markdown('### Model Description') |
|
st.text_area("Provide a 1-2 sentence summary of what this model is.", help="The model description provides basic details about the model. This includes the architecture, version, if it was introduced in a paper, if an original implementation is available, the author, and general information about the model. Any copyright should be attributed here. General information about training procedures, parameters, and important disclaimers can also be mentioned in this section.", key=persist('model_description')) |
|
|
|
left, right = st.columns([2,6], gap="small") |
|
with left: |
|
st.write("\n") |
|
st.write("\n") |
|
st.markdown('### Developed By:') |
|
st.write("\n") |
|
st.write("\n") |
|
|
|
st.markdown('### Shared By [optional]:') |
|
st.write("\n") |
|
st.write("\n") |
|
st.markdown('### Model Type:') |
|
st.write("\n") |
|
st.write("\n") |
|
st.markdown('### License:') |
|
with right: |
|
st.write("\n") |
|
st.write("\n") |
|
st.text_input("",help="Developed By work", key=persist("Model_developers")) |
|
st.write("\n") |
|
st.write("\n") |
|
|
|
st.text_input("",help="Shared By work",key=persist("shared_by")) |
|
st.text_input("",help="Model Type work") |
|
|
|
st.selectbox("",[""] + list(license_map.values()), help="Licenses work", key=persist("license")) |
|
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
load_widget_state() |
|
main() |
