Add application file

.gitattributes

@@ -32,4 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.xz filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
+*.keras filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text

Pokemons.txt

@@ -0,0 +1,151 @@
+Abra
+Aerodactyl
+Alakazam
+Arbok
+Arcanine
+Articuno
+Beedrill
+Bellsprout
+Blastoise
+Bulbasaur
+Butterfree
+Caterpie
+Chansey
+Charizard
+Charmander
+Charmeleon
+Clefable
+Clefairy
+Cloyster
+Cubone
+Dewgong
+Diglett
+Ditto
+Dodrio
+Doduo
+Dragonair
+Dragonite
+Dratini
+Drowzee
+Dugtrio
+Eevee
+Ekans
+Electabuzz
+Electrode
+Exeggcute
+Exeggutor
+Farfetch
+Fearow
+Flareon
+Gastly
+Gengar
+Geodude
+Gloom
+Golbat
+Goldeen
+Golduck
+Golem
+Graveler
+Grimer
+Growlithe
+Gyarados
+Haunter
+Hitmonchan
+Hitmonlee
+Horsea
+Hypno
+Ivysaur
+Jigglypuff
+Jolteon
+Jynx
+Kabuto
+Kabutops
+Kadabra
+Kakuna
+Kangaskhan
+Kingler
+Koffing
+Krabby
+Lapras
+Lickitung
+Machamp
+Machoke
+Machop
+Magikarp
+Magmar
+Magnemite
+Magneton
+Mankey
+Marowak
+Meowth
+Metapod
+Mew
+Mewtwo
+Moltres
+Mr-Mime
+Muk
+Nidoking
+Nidoqueen
+Nidoran♀
+Nidoran♂
+Nidorina
+Nidorino
+Ninetales
+Oddish
+Omanyte
+Omastar
+Onix
+Paras
+Parasect
+Persian
+Pidgeot
+Pidgeotto
+Pidgey
+Pikachu
+Pinsir
+Poliwag
+Poliwhirl
+Poliwrath
+Ponyta
+Porygon
+Primeape
+Psyduck
+Raichu
+Rapidash
+Raticate
+Rattata
+Rhydon
+Rhyhorn
+Sandshrew
+Sandslash
+Scyther
+Seadra
+Seaking
+Seel
+Shellder
+Slowbro
+Slowpoke
+Snorlax
+Spearow
+Squirtle
+Starmie
+Staryu
+Tangela
+Tauros
+Tentacool
+Tentacruel
+Vaporeon
+Venomoth
+Venonat
+Venusaur
+Victreebel
+Vileplume
+Voltorb
+Vulpix
+Wartortle
+Weedle
+Weepinbell
+Weezing
+Wigglytuff
+Zapdos
+Zubat

README.md

@@ -1,12 +1,11 @@
 ---
 title: Pokedex
-emoji: 🌖
-colorFrom: purple
+emoji: ⚡
+colorFrom: red
 colorTo: gray
 sdk: gradio
 sdk_version: 5.25.2
 app_file: app.py
 pinned: false
+short_description: Pokémon first gen classifier
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,52 @@
+import gradio as gr
+import keras
+from keras.src.applications.densenet import preprocess_input
+import numpy as np
+
+pokedex = keras.saving.load_model("pokedex.keras")
+
+with open('Pokemons.txt', 'r') as f:
+  class_labels = f.read().splitlines()
+
+
+def classify_pokemon(image):
+
+  img = image.resize((224, 224))
+
+  x = keras.utils.img_to_array(img)
+  x = np.expand_dims(x, axis=0)
+  x = preprocess_input(x)
+
+  preds = pokedex.predict(x)
+
+  top_indices = preds[0].argsort()[-3:][::-1]
+
+  results = {class_labels[i]: float(preds[0][i]) for i in top_indices}
+  return results
+
+
+title = "Pokedex"
+description = "Pokémon first gen classifier"
+
+examples = [
+    'examples/Pikachu.png',
+    'examples/Charmander.png',
+    'examples/Squirtle.png',
+    'examples/Bulbasaur.png',
+    'examples/Caterpie.png',
+    'examples/Cloyster.png',
+    'examples/Gengar.png',
+    'examples/Porygon.png',
+    'examples/Rapidash.png',
+    'examples/Slowpoke.png',
+]
+intf = gr.Interface(
+    fn=classify_pokemon,
+    inputs=gr.Image(type='pil', label="Upload a Pokémon image"),
+    outputs=gr.Label(num_top_classes=3, label="Prediction"),
+    examples=examples,
+    title=title,
+    description=description,
+)
+
+intf.launch(inline=False)

pokedex.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e5f5e771e137c6b5803a751140f5a545e9e9cf542a01a2d7475e73937ed31be4
+size 87375811

requirements.txt

@@ -0,0 +1,77 @@
+absl-py==2.2.2
+aiofiles==24.1.0
+annotated-types==0.7.0
+anyio==4.9.0
+astunparse==1.6.3
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+exceptiongroup==1.2.2
+fastapi==0.115.12
+ffmpy==0.5.0
+filelock==3.18.0
+flatbuffers==25.2.10
+fsspec==2025.3.2
+gast==0.6.0
+google-pasta==0.2.0
+gradio==5.25.2
+gradio_client==1.8.0
+groovy==0.1.2
+grpcio==1.71.0
+h11==0.14.0
+h5py==3.13.0
+httpcore==1.0.8
+httpx==0.28.1
+huggingface-hub==0.30.2
+idna==3.10
+Jinja2==3.1.6
+keras==3.9.2
+libclang==18.1.1
+Markdown==3.8
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+mdurl==0.1.2
+ml_dtypes==0.5.1
+namex==0.0.8
+numpy==2.1.3
+opt_einsum==3.4.0
+optree==0.15.0
+orjson==3.10.16
+packaging==24.2
+pandas==2.2.3
+pillow==11.2.1
+protobuf==5.29.4
+pydantic==2.11.3
+pydantic_core==2.33.1
+pydub==0.25.1
+Pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.2
+PyYAML==6.0.2
+requests==2.32.3
+rich==14.0.0
+ruff==0.11.5
+safehttpx==0.1.6
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+starlette==0.46.2
+tensorboard==2.19.0
+tensorboard-data-server==0.7.2
+tensorflow==2.19.0
+tensorflow-io-gcs-filesystem==0.31.0
+termcolor==3.0.1
+tomlkit==0.13.2
+tqdm==4.67.1
+typer==0.15.2
+typing-inspection==0.4.0
+typing_extensions==4.13.2
+tzdata==2025.2
+urllib3==2.4.0
+uvicorn==0.34.1
+websockets==15.0.1
+Werkzeug==3.1.3
+wrapt==1.17.2