TroglodyteDerivations commited on
Commit
d183073
1 Parent(s): f0c3a98

Updated notation lines 12-16 via labels variable declaration : # Define the Pokémon labels # Although, labels are set with 3 Pokémon -> One can leverage the remaining 715 Pokémon # Provides the functionality -> But, the hardcoded list of three Pokémon, is not correct. # On the next variant [mutation] of the Pokémon classifier jettison obverse the actual model's configuration # yielding mapping from class indices to labels. Then, foment the model's predictions on the Pokémon.

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -9,8 +9,11 @@ model_name = "imjeffhi/pokemon_classifier"
9
  model = AutoModelForImageClassification.from_pretrained(model_name)
10
  feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
11
 
12
- # Define the Pokemon labels
13
  # Although, labels are set with 3 Pokémon -> One can leverage the remaining 715 Pokémon
 
 
 
14
  labels = ['Jolteon', 'Kakuna', 'Mr. Mime']
15
 
16
  # Function to preprocess the image
 
9
  model = AutoModelForImageClassification.from_pretrained(model_name)
10
  feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
11
 
12
+ # Define the Pokémon labels
13
  # Although, labels are set with 3 Pokémon -> One can leverage the remaining 715 Pokémon
14
+ # Provides the functionality -> But, the hardcoded list of three Pokémon, is not correct.
15
+ # On the next variant [mutation] of the Pokémon classifier jettison obverse the actual model's configuration
16
+ # yielding mapping from class indices to labels. Then, foment the model's predictions on the Pokémon.
17
  labels = ['Jolteon', 'Kakuna', 'Mr. Mime']
18
 
19
  # Function to preprocess the image