tombio commited on
Commit
316872d
1 Parent(s): 2fcd80e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ model = load_model_from_config(config, ckpt, device=device, verbose=False)
28
  model = model.to(device).half()
29
 
30
  clip_model, preprocess = clip.load("ViT-L/14", device=device)
31
- gender_learn = load_learner('model.pkl')
32
  gender_labels = gender_learn.dls.vocab
33
 
34
  n_inputs = 5
 
28
  model = model.to(device).half()
29
 
30
  clip_model, preprocess = clip.load("ViT-L/14", device=device)
31
+ gender_learn = load_learner('gender_model.pkl')
32
  gender_labels = gender_learn.dls.vocab
33
 
34
  n_inputs = 5