Skip to content

2.2: Upload your model

Zhikai Li edited this page Aug 6, 2024 · 9 revisions

You can upload the model to your Huggingface repository and then contribute it to SaprotHub.

Config

You need to add some description for your model:

  • name: The name of your model.
  • description: The description of your model (which task is your model used for).
  • label_meanings:
    • For classification model, please provide detailed information about the meanings of all labels.
    • For regression model, please provide the numerical range of value.

Example

For classification models:

Parameter Value
name Subcellular_Localization
description This model is used for the Subcellular Localization Classification Task. It takes a protein sequence as input and outputs which of the 10 categories the protein belongs to.
label_meanings Nucleus, Cytoplasm, Extracellular, Mitochondrion, Cell.membrane, Endoplasmic.reticulum, Plastid, Golgi.apparatus, Lysosome/Vacuole, Peroxisome

For regression models:

Parameter Value
name Thermostability
description This model is used for the Thermostability Regression Task. It takes a protein sequence as input and outputs the thermostability of the protein.
label_meanings Label corresponds to the protein melting temperature (Tm) normalized using the Min-Max normalization method.

You can also edit the README.md to provide more information in the model card, such as Dataset description, Performance and so on.

Instruction

  1. Click run button, Find your token and Login Huggingface
  2. Complete model card config and then click run button to upload
  3. Check your model repo