nicolas-dufour commited on
Commit
216bfa4
·
verified ·
1 Parent(s): d6c16d6

Update app.py

Browse files

change order of defaults

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ PROJECT_ROOT = Path(__file__).parent.parent.absolute()
20
  CHECKPOINT_DIR = PROJECT_ROOT / "checkpoints"
21
 
22
  MODEL_NAMES = {
23
- "PLONK_YFCC": "nicolas-dufour/PLONK_YFCC",
24
  "PLONK_OSV_5M": "nicolas-dufour/PLONK_OSV_5M",
 
25
  "PLONK_iNaturalist": "nicolas-dufour/PLONK_iNaturalist",
26
  }
27
 
@@ -207,7 +207,7 @@ def main():
207
  model_name = st.selectbox(
208
  "🤖 Select Model",
209
  options=MODEL_NAMES.keys(),
210
- index=0, # Default to YFCC
211
  help="Choose which PLONK model variant to use for prediction.",
212
  )
213
 
 
20
  CHECKPOINT_DIR = PROJECT_ROOT / "checkpoints"
21
 
22
  MODEL_NAMES = {
 
23
  "PLONK_OSV_5M": "nicolas-dufour/PLONK_OSV_5M",
24
+ "PLONK_YFCC": "nicolas-dufour/PLONK_YFCC",
25
  "PLONK_iNaturalist": "nicolas-dufour/PLONK_iNaturalist",
26
  }
27
 
 
207
  model_name = st.selectbox(
208
  "🤖 Select Model",
209
  options=MODEL_NAMES.keys(),
210
+ index=0, # Default to OSV
211
  help="Choose which PLONK model variant to use for prediction.",
212
  )
213