Hugging Face's logo

Spaces:
vdcapriles
/
Pokedex_classifier
Sleeping

App Files Community
victor commited on
Commit
7095d4d
·
1 Parent(s): 4ee79c9

added txt

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +0 -0
app.py CHANGED
@@ -41,7 +41,7 @@ def get_y(item):
41
  return class_names[item['labels']] # Map label index to class name
42
 
43
  # Load the model
44
- learn = load_learner('poke_model.pkl', cpu=False) # Ensure correct path to poke_model.pkl
45
 
46
  # Categories are derived from the vocabulary of the dataloader used during training
47
  categories = learn.dls.vocab # These are the class labels used in the trained model
 
41
  return class_names[item['labels']] # Map label index to class name
42
 
43
  # Load the model
44
+ learn = load_learner('poke_model.pkl')
45
 
46
  # Categories are derived from the vocabulary of the dataloader used during training
47
  categories = learn.dls.vocab # These are the class labels used in the trained model
requirements.txt ADDED
Binary file (3.76 kB). View file