Add application files

README.md

@@ -8,6 +8,8 @@ sdk_version: 3.45.2
 app_file: app.py
 pinned: false
 license: mit
+models:
+- Okkoman/PokeFace
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,33 @@
+__all__ = ['modelname', 'pokemon_types', 'pokemon_types_en', 'pokemon_types_fr', 'path', 'learn_inf', 'image', 'label',
+           'examples', 'intf', 'classify_image']
+
+from fastai.vision.all import *
+
+modelname = 'model.pkl'
+
+pokemon_types = pd.read_csv("pokemongen1patch.csv", nrows=20)
+pokemon_types_en = pokemon_types['en']
+pokemon_types_fr = pokemon_types['fr']
+
+path = Path()
+learn_inf = load_learner(path/modelname)
+learn_inf.dls.vocab
+
+import gradio as gr
+
+def classify_image(img):
+    prob_threshold = 0.8
+    pred,pred_idx,probs = learn_inf.predict(img)
+    index = pokemon_types_en[pokemon_types_en == pred].index[0]
+    pred_fr = pokemon_types_fr.iloc[index]
+    if probs[pred_idx] > prob_threshold:
+        return f"{pred}(en) - {pred_fr}(fr) - {probs[pred_idx]*100:.0f}%"
+    else:
+        return 'unknown'
+
+image = gr.inputs.Image(shape=(192,192))
+label = gr.outputs.Label()
+examples = ['pikachu.webp', 'bulbizarre.jpg', 'tortank.png']
+
+intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+intf.launch(inline=False)

pokemongen1patch.csv

@@ -0,0 +1,154 @@
+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,
+
+