using model_gen1

app.py

@@ -1,25 +1,26 @@
-# AUTOGENERATED! DO NOT EDIT! File to edit: ../pokemonclassifier.ipynb.
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../app.ipynb.
 
 # %% auto 0
-__all__ = ['modelname', 'imagespath', 'pokemon_types', 'pokemon_types_en', 'learn_inf', 'lang', 'title', 'description', 'unknown',
-           'prob_threshold', 'classify_image']
+__all__ = ['generation', 'modelname', 'pokemon_types', 'pokemon_types_en', 'examplespath', 'learn_inf', 'lang', 'title',
+           'description', 'unknown', 'prob_threshold', 'classify_image']
 
-# %% ../pokemonclassifier.ipynb 3
+# %% ../app.ipynb 3
 import pandas as pd
 
-modelname = 'model.pkl'
-imagespath = 'images/'
-
-pokemon_types = pd.read_csv("pokemongen1patch.csv")
+generation = 1
+modelname = f'model_gen{generation}.pkl'
+pokemon_types = pd.read_csv(f'pokemongen{generation}.csv')
 pokemon_types_en = pokemon_types['en']
 
-# %% ../pokemonclassifier.ipynb 27
+examplespath = 'images/'
+
+# %% ../app.ipynb 7
 from huggingface_hub import hf_hub_download
 from fastai.learner import load_learner
 
 learn_inf = load_learner(hf_hub_download("Okkoman/PokeFace151", modelname))
 
-# %% ../pokemonclassifier.ipynb 30
+# %% ../app.ipynb 9
 import gradio as gr
 
 lang = 'en'
@@ -57,7 +58,7 @@ with gr.Blocks() as demo:
             fn=classify_image, 
             inputs=gr.inputs.Image(shape=(192,192)), 
             outputs=gr.outputs.Label(), 
-            examples=imagespath, 
+            examples=examplespath, 
             allow_flagging='auto')
 
 demo.launch(inline=False)

pokemongen1.csv

@@ -0,0 +1,152 @@
+en,fr,jp,num,type1,type2
+Bulbasaur,Bulbizarre,Fushigidane,0001,Plante,Poison
+Ivysaur,Herbizarre,Fushigisou,0002,Plante,Poison
+Venusaur,Florizarre,Fushigibana,0003,Plante,Poison
+Charmander,Salamèche,Hitokage,0004,Feu,
+Charmeleon,Reptincel,Lizardo,0005,Feu,
+Charizard,Dracaufeu,Lizardon,0006,Feu,Vol
+Squirtle,Carapuce,Zenigame,0007,Eau,
+Wartortle,Carabaffe,Kameil,0008,Eau,
+Blastoise,Tortank,Kamex,0009,Eau,
+Caterpie,Chenipan,Caterpie,0010,Insecte,
+Metapod,Chrysacier,Transel,0011,Insecte,
+Butterfree,Papilusion,Butterfree,0012,Insecte,Vol
+Weedle,Aspicot,Beedle,0013,Insecte,Poison
+Kakuna,Coconfort,Cocoon,0014,Insecte,Poison
+Beedrill,Dardargnan,Spear,0015,Insecte,Poison
+Pidgey,Roucool,Poppo,0016,Normal,Vol
+Pidgeotto,Roucoups,Pigeon,0017,Normal,Vol
+Pidgeot,Roucarnage,Pigeot,0018,Normal,Vol
+Rattata,Rattata,Koratta,0019,Normal,
+Raticate,Rattatac,Ratta,0020,Normal,
+Spearow,Piafabec,Onisuzume,0021,Normal,Vol
+Fearow,Rapasdepic,Onidrill,0022,Normal,Vol
+Ekans,Abo,Arbo,0023,Poison,
+Arbok,Arbok,Arbok,0024,Poison,
+Pikachu,Pikachu,Pikachu,0025,Électrik,
+Raichu,Raichu,Raichu,0026,Électrik,
+Sandshrew,Sabelette,Sand,0027,Sol,
+Sandslash,Sablaireau,Sandpan,0028,Sol,
+Nidoran♀,Nidoran♀,Nidoran♀,0029,Poison,
+Nidorina,Nidorina,Nidorina,0030,Poison,
+Nidoqueen,Nidoqueen,Nidoqueen,0031,Poison,Sol
+Nidoran♂,Nidoran♂,Nidoran♂,0032,Poison,
+Nidorino,Nidorino,Nidorino,0033,Poison,
+Nidoking,Nidoking,Nidoking,0034,Poison,Sol
+Clefairy,Mélofée,Pippi,0035,Fée,
+Clefable,Mélodelfe,Pixy,0036,Fée,
+Vulpix,Goupix,Rokon,0037,Feu,
+Ninetales,Feunard,Kyukon,0038,Feu,
+Jigglypuff,Rondoudou,Purin,0039,Normal,Fée
+Wigglytuff,Grodoudou,Pukurin,0040,Normal,Fée
+Zubat,Nosferapti,Zubat,0041,Poison,Vol
+Golbat,Nosferalto,Golbat,0042,Poison,Vol
+Oddish,Mystherbe,Nazonokusa,0043,Plante,Poison
+Gloom,Ortide,Kusaihana,0044,Plante,Poison
+Vileplume,Rafflesia,Ruffresia,0045,Plante,Poison
+Paras,Paras,Paras,0046,Insecte,Plante
+Parasect,Parasect,Parasect,0047,Insecte,Plante
+Venonat,Mimitoss,Kongpang,0048,Insecte,Poison
+Venomoth,Aéromite,Morphon,0049,Insecte,Poison
+Diglett,Taupiqueur,Digda,0050,Sol,
+Dugtrio,Triopikeur,Dugtrio,0051,Sol,
+Meowth,Miaouss,Nyarth,0052,Normal,
+Persian,Persian,Persian,0053,Normal,
+Psyduck,Psykokwak,Koduck,0054,Eau,
+Golduck,Akwakwak,Golduck,0055,Eau,
+Mankey,Férosinge,Mankey,0056,Combat,
+Primeape,Colossinge,Okorizaru,0057,Combat,
+Growlithe,Caninos,Gardie,0058,Feu,
+Arcanine,Arcanin,Windie,0059,Feu,
+Poliwag,Ptitard,Nyoromo,0060,Eau,
+Poliwhirl,Têtarte,Nyorozo,0061,Eau,
+Poliwrath,Tartard,Nyorobon,0062,Eau,Combat
+Abra,Abra,Casey,0063,Psy,
+Kadabra,Kadabra,Yungerer,0064,Psy,
+Alakazam,Alakazam,Foodin,0065,Psy,
+Machop,Machoc,Wanriky,0066,Combat,
+Machoke,Machopeur,Goriky,0067,Combat,
+Machamp,Mackogneur,Kairiky,0068,Combat,
+Bellsprout,Chétiflor,Madatsubomi,0069,Plante,Poison
+Weepinbell,Boustiflor,Utsudon,0070,Plante,Poison
+Victreebel,Empiflor,Utsubot,0071,Plante,Poison
+Tentacool,Tentacool,Menokurage,0072,Eau,Poison
+Tentacruel,Tentacruel,Dokukurage,0073,Eau,Poison
+Geodude,Racaillou,Isitsubute,0074,Roche,Sol
+Graveler,Gravalanch,Golone,0075,Roche,Sol
+Golem,Grolem,Golonya,0076,Roche,Sol
+Ponyta,Ponyta,Ponyta,0077,Feu,
+Rapidash,Galopa,Gallop,0078,Feu,
+Slowpoke,Ramoloss,Yadon,0079,Eau,Psy
+Slowbro,Flagadoss,Yadoran,0080,Eau,Psy
+Magnemite,Magnéti,Coil,0081,Électrik,Acier
+Magneton,Magnéton,Rarecoil,0082,Électrik,Acier
+Farfetch'd,Canarticho,Kamonegi,0083,Normal,Vol
+Doduo,Doduo,Dodo,0084,Normal,Vol
+Dodrio,Dodrio,Dodorio,0085,Normal,Vol
+Seel,Otaria,Pawou,0086,Eau,
+Dewgong,Lamantine,Jugon,0087,Eau,Glace
+Grimer,Tadmorv,Betbeter,0088,Poison,
+Muk,Grotadmorv,Betbeton,0089,Poison,
+Shellder,Kokiyas,Shellder,0090,Eau,
+Cloyster,Crustabri,Parshen,0091,Eau,Glace
+Gastly,Fantominus,Ghos,0092,Spectre,Poison
+Haunter,Spectrum,Ghost,0093,Spectre,Poison
+Gengar,Ectoplasma,Gangar,0094,Spectre,Poison
+Onix,Onix,Iwark,0095,Roche,Sol
+Drowzee,Soporifik,Sleepe,0096,Psy,
+Hypno,Hypnomade,Sleeper,0097,Psy,
+Krabby,Krabby,Crab,0098,Eau,
+Kingler,Krabboss,Kingler,0099,Eau,
+Voltorb,Voltorbe,Biriridama,0100,Électrik,
+Electrode,Électrode,Marumine,0101,Électrik,
+Exeggcute,Noeunoeuf,Tamatama,0102,Plante,Psy
+Exeggutor,Noadkoko,Nassy,0103,Plante,Psy
+Cubone,Osselait,Karakara,0104,Sol,
+Marowak,Ossatueur,Garagara,0105,Sol,
+Hitmonlee,Kicklee,Sawamular,0106,Combat,
+Hitmonchan,Tygnon,Ebiwalar,0107,Combat,
+Lickitung,Excelangue,Beroringa,0108,Normal,
+Koffing,Smogo,Dogars,0109,Poison,
+Weezing,Smogogo,Matadogas,0110,Poison,
+Rhyhorn,Rhinocorne,Sihorn,0111,Sol,Roche
+Rhydon,Rhinoféros,Sidon,0112,Sol,Roche
+Chansey,Leveinard,Lucky,0113,Normal,
+Tangela,Saquedeneu,Monjara,0114,Plante,
+Kangaskhan,Kangourex,Garura,0115,Normal,
+Horsea,Hypotrempe,Tattu,0116,Eau,
+Seadra,Hypocéan,Seadra,0117,Eau,
+Goldeen,Poissirène,Tosakinto,0118,Eau,
+Seaking,Poissoroy,Azumao,0119,Eau,
+Staryu,Stari,Hitodeman,0120,Eau,
+Starmie,Staross,Starmie,0121,Eau,Psy
+Mr. Mime,M. Mime,Barrierd,0122,Psy,Fée
+Scyther,Insécateur,Strike,0123,Insecte,Vol
+Jynx,Lippoutou,Rougela,0124,Glace,Psy
+Electabuzz,Élektek,Eleboo,0125,Électrik,
+Magmar,Magmar,Boober,0126,Feu,
+Pinsir,Scarabrute,Kailios,0127,Insecte,
+Tauros,Tauros,Kentauros,0128,Normal,
+Magikarp,Magicarpe,Koiking,0129,Eau,
+Gyarados,Léviator,Gyarados,0130,Eau,Vol
+Lapras,Lokhlass,Laplace,0131,Eau,Glace
+Ditto,Métamorph,Metamon,0132,Normal,
+Eevee,Évoli,Eievui,0133,Normal,
+Vaporeon,Aquali,Showers,0134,Eau,
+Jolteon,Voltali,Thunders,0135,Électrik,
+Flareon,Pyroli,Booster,0136,Feu,
+Porygon,Porygon,Porygon,0137,Normal,
+Omanyte,Amonita,Omnite,0138,Roche,Eau
+Omastar,Amonistar,Omstar,0139,Roche,Eau
+Kabuto,Kabuto,Kabuto,0140,Roche,Eau
+Kabutops,Kabutops,Kabutops,0141,Roche,Eau
+Aerodactyl,Ptéra,Ptera,0142,Roche,Vol
+Snorlax,Ronflex,Kabigon,0143,Normal,
+Articuno,Artikodin,Freezer,0144,Glace,Vol
+Zapdos,Électhor,Thunder,0145,Électrik,Vol
+Moltres,Sulfura,Fire,0146,Feu,Vol
+Dratini,Minidraco,Miniryu,0147,Dragon,
+Dragonair,Draco,Hakuryu,0148,Dragon,
+Dragonite,Dracolosse,Kairyu,0149,Dragon,Vol
+Mewtwo,Mewtwo,Mewtwo,0150,Psy,
+Mew,Mew,Mew,0151,Psy,

pokemongen1patch.csv

@@ -1,154 +0,0 @@
-en,fr,jp,num,type1,type2
-Bulbasaur,Bulbizarre,フシギダネ,001,Grass,Poison
-Ivysaur,Herbizarre,フシギソウ,002,Grass,Poison
-Venusaur,Florizarre,フシギバナ,003,Grass,Poison
-Charmander,Salamèche,ヒトカゲ,004,Fire,
-Charmeleon,Reptincel,リザード,005,Fire,
-Charizard,Dracaufeu,リザードン,006,Fire,Flying
-Squirtle,Carapuce,ゼニガメ,007,Water,
-Wartortle,Carabaffe,カメール,008,Water,
-Blastoise,Tortank,カメックス,009,Water,
-Caterpie,Chenipan,キャタピー,010,Bug,
-Metapod,Chrysacier,トランセル,011,Bug,
-Butterfree,Papilusion,バタフリー,012,Bug,Flying
-Weedle,Aspicot,ビードル,013,Bug,Poison
-Kakuna,Coconfort,コクーン,014,Bug,Poison
-Beedrill,Dardargnan,スピアー,015,Bug,Poison
-Pidgey,Roucool,ポッポ,016,Normal,Flying
-Pidgeotto,Roucoups,ピジョン,017,Normal,Flying
-Pidgeot,Roucarnage,ピジョット,018,Normal,Flying
-Rattata,Rattata,コラッタ,019,Normal,
-Raticate,Rattatac,ラッタ,020,Normal,
-Spearow,Piafabec,オニスズメ,021,Normal,Flying
-Fearow,Rapasdepic,オニドリル,022,Normal,Flying
-Ekans,Abo,アーボ,023,Poison,
-Arbok,Arbok,アーボック,024,Poison,
-Pikachu,Pikachu,ピカチュウ,025,Electric,
-Raichu,Raichu,ライチュウ,026,Electric,
-Sandshrew,Sabelette,サンド,027,Ground,
-Sandslash,Sablaireau,サンドパン,028,Ground,
-Nidoran♀,Nidoran♀,ニドラン♀,029,Poison,
-Nidorina,Nidorina,ニドリーナ,030,Poison,
-Nidoqueen,Nidoqueen,ニドクイン,031,Poison,Ground
-Nidoran♂,Nidoran♂,ニドラン♂,032,Poison,
-Nidorino,Nidorino,ニドリーノ,033,Poison,
-Nidoking,Nidoking,ニドキング,034,Poison,Ground
-Clefairy,Mélofée,ピッピ,035,Fairy,
-Clefable,Mélodelfe,ピクシー,036,Fairy,
-Vulpix,Goupix,ロコン,037,Fire,
-Ninetales,Feunard,キュウコン,038,Fire,
-Jigglypuff,Rondoudou,プリン,039,Normal,Fairy
-Wigglytuff,Grodoudou,プクリン,040,Normal,Fairy
-Zubat,Nosferapti,ズバット,041,Poison,Flying
-Golbat,Nosferalto,ゴルバット,042,Poison,Flying
-Oddish,Mystherbe,ナゾノクサ,043,Grass,Poison
-Gloom,Ortide,クサイハナ,044,Grass,Poison
-Vileplume,Rafflesia,ラフレシア,045,Grass,Poison
-Paras,Paras,パラス,046,Bug,Grass
-Parasect,Parasect,パラセクト,047,Bug,Grass
-Venonat,Mimitoss,コンパン,048,Bug,Poison
-Venomoth,Aéromite,モルフォン,049,Bug,Poison
-Digglet,Taupiqueur,ディグダ,050,Ground,
-Dugtrio,Triopikeur,ダグトリオ,051,Ground,
-Meowth,Miaouss,ニャース,052,Normal,
-Persian,Persian,ペルシアン,053,Normal,
-Psyduck,Psykokwak,コダック,054,Water,
-Golduck,Akwakwak,ゴルダック,055,Water,
-Mankey,Férosinge,マンキー,056,Fighting,
-Primeape,Colossinge,オコリザル,057,Fighting,
-Growlithe,Caninos,ガーディ,058,Fire,
-Arcanine,Arcanin,ウインディ,059,Fire,
-Poliwag,Ptitard,ニョロモ,060,Water,
-Poliwhirl,Têtarte,ニョロゾ,061,Water,
-Poliwrath,Tartard,ニョロボン,062,Water,Fighting
-Abra,Abra,ケーシィ,063,Psychic,
-Kadabra,Kadabra,ユンゲラー,064,Psychic,
-Alakazam,Alakazam,フーディン,065,Psychic,
-Machop,Machoc,ワンリキ,066,Fighting,
-Machoke,Machopeur,ゴーリキー,067,Fighting,
-Machamp,Mackogneur,カイリキー,068,Fighting,
-Bellsprout,Chétiflor,マダツボミ,069,Grass,Poison
-Weepinbell,Boustiflor,マダツホソウ,070,Grass,Poison
-Victreebel,Empiflor,ウツドン,071,Grass,Poison
-Tentacool,Tentacool,メノクラゲ,072,Water,Poison
-Tentacruel,Tentacruel,ドククラゲ,073,Water,Poison
-Geodude,Racaillou,イシツブテ,074,Rock,Ground
-Graveler,Gravalanch,ゴローン,075,Rock,Ground
-Golem,Grolem,ゴローニャ,076,Rock,Ground
-Ponyta,Galopa,ポニータ,077,Fire,
-Rapidash,Galopa,ギャロップ,078,Fire,
-Slowpoke,Ramoloss,ヤドン,079,Water,Psychic
-Slowbro,Flagadoss,ヤドラン,080,Water,Psychic
-Magnemite,Magnéti,コイル,081,Electric,Steel
-Magneton,Magnéton,レアコイル,082,Electric,Steel
-Farfetch'd,Canarticho,カモネギ,083,Normal,Flying
-Doduo,Doduo,ドードー,084,Normal,Flying
-Dodrio,Dodrio,ドードリオ,085,Normal,Flying
-Seel,Otaria,パウワウ,086,Water,
-Dewgong,Lamantine,ジュゴン,087,Water,Ice
-Grimer,Tadmorv,ベトベター,088,Poison,
-Muk,Grotadmorv,ベトベトン,089,Poison,
-Shellder,Kokiyas,シェルダー,090,Water,
-Cloyster,Crustabri,パルシェン,091,Water,Ice
-Gastly,Fantominus,ゴース,092,Ghost,Poison
-Haunter,Spectrum,ゴースト,093,Ghost,Poison
-Gengar,Ectoplasma,ゲンガー,094,Ghost,Poison
-Onix,Onix,イワーク,095,Rock,Ground
-Drowzee,Soporifik,スリープ,096,Psychic,
-Hypno,Hypnomade,スリーパー,097,Psychic,
-Krabby,Krabby,クラブ,098,Water,
-Kingler,Krabboss,キングラー,099,Water,
-Voltorb,Voltorbe,ビリリダマ,100,Electric,
-Electrode,Électrode,マルマイン,101,Electric,
-Exeggcute,Noeunoeuf,タマタマ,102,Grass,Psychic
-Exeggutor,Noadkoko,ナッシー,103,Grass,Psychic
-Cubone,Osselait,カラカラ,104,Ground,
-Marowak,Ossatueur,ガラガラ,105,Ground,
-Hitmonlee,Kicklee,��ワムラー,106,Fighting,
-Hitmonchan,Tygnon,エビワラー,107,Fighting,
-Lickitung,Excelangue,ベロリンガ,108,Normal,
-Koffing,Smogo,ドガース,109,Poison,
-Weezing,Smogogo,マタドガス,110,Poison,
-Rhyhorn,Rhinocorne,サイホーン,111,Ground,Rock
-Rhydon,Rhinoféros,サイドン,112,Ground,Rock
-Chansey,Leveinard,ラッキー,113,Normal,
-Tangela,Saquedeneu,モンジャラ,114,Grass,
-Kangaskhan,Kangourex,ガルーラ,115,Normal,
-Horsea,Hypotrempe,タッツー,116,Water,
-Seadra,Hypocéan,シードラ,117,Water,
-Goldeen,Poissirène,トサキント,118,Water,
-Seaking,Poissoroy,アズマオウ,119,Water,
-Staryu,Stari,ヒトデマン,120,Water,
-Starmie,Staross,スターミー,121,Water,Psychic
-Mr. Mime,M. Mime,バリヤード,122,Psychic,Fairy
-Scyther,Insecateur,ストライク,123,Bug,Flying
-Jynx,Lippoutou,ルージュラ,124,Ice,Psychic
-Electabuzz,Élektek,エレブー,125,Electric,
-Magmar,Magmar,ブーバー,126,Fire,
-Pinsir,Scarabrute,カイロス,127,Bug,
-Tauros,Tauros,ケンタロス,128,Normal,
-Magikarp,Magicarpe,コイキング,129,Water,
-Gyarados,Léviator,ギャラドス,130,Water,Flying
-Lapras,Lokhlass,ラプラス,131,Water,Ice
-Ditto,Métamorph,メタモン,132,Normal,
-Eevee,Évoli,イーブイ,133,Normal,
-Vaporeon,Aquali,シャワーズ,134,Water,
-Jolteon,Voltali,サンダース,135,Electric,
-Flareon,Pyroli,ブースター,136,Fire,
-Porygon,Porygon,ポリゴン,137,Normal,
-Omanyte,Amonita,オムナイト,138,Rock,Water
-Omastar,Amonistar,オムスター,139,Rock,Water
-Kabuto,Kabuto,カブト,140,Rock,Water
-Kabutops,Kabutops,カブトプス,141,Rock,Water
-Aerodactyl,Ptéra,プテラ,142,Rock,Flying
-Snorlax,Ronflex,カビゴン,143,Normal,
-Articuno,Artikodin,フリーザー,144,Ice,Flying
-Zapdos,Électhor,サンダー,145,Electric,Flying
-Moltres,Sulfura,ファイヤー,146,Fire,Flying
-Dratini,Minidraco,ミニリュウ,147,Dragon,
-Dragonair,Draco,ハクリュー,148,Dragon,
-Dragonite,Dracolosse,カイリュー,149,Dragon,Flying
-Mewtwo,Mewtwo,ミュウツー,150,Psychic,
-Mew,Mew,ミュウ,151,Psychic,
-
-