ccclllwww commited on
Commit
a83b95d
·
verified ·
1 Parent(s): 3008e25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def load_gender_pipeline():
33
  @st.cache_resource
34
  def load_age_pipeline():
35
  """Initialize and cache the age image classification pipeline."""
36
- return pipeline("image-classification", model="akashmaggon/vit-base-age-classification", use_fast=True)
37
 
38
  # Preload all models
39
  smoke_pipeline = load_smoke_pipeline()
 
33
  @st.cache_resource
34
  def load_age_pipeline():
35
  """Initialize and cache the age image classification pipeline."""
36
+ return pipeline("image-classification", model="cledoux42/Age_Classify_v001", use_fast=True)
37
 
38
  # Preload all models
39
  smoke_pipeline = load_smoke_pipeline()