put examples in images dir and change to model 151

app.py

@@ -1,13 +1,14 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../pokemonclassifier.ipynb.
 
 # %% auto 0
-__all__ = ['modelname', 'pokemon_types', 'pokemon_types_en', 'learn_inf', 'imagespath', 'lang', 'title', 'description', 'unknown',
+__all__ = ['modelname', 'imagespath', 'pokemon_types', 'pokemon_types_en', 'learn_inf', 'lang', 'title', 'description', 'unknown',
            'prob_threshold', 'classify_image']
 
 # %% ../pokemonclassifier.ipynb 3
 import pandas as pd
 
 modelname = 'model.pkl'
+imagespath = 'images/'
 
 pokemon_types = pd.read_csv("pokemongen1patch.csv")
 pokemon_types_en = pokemon_types['en']
@@ -16,14 +17,13 @@ pokemon_types_en = pokemon_types['en']
 from huggingface_hub import hf_hub_download
 from fastai.learner import load_learner
 
-learn_inf = load_learner(hf_hub_download("Okkoman/PokeFace", modelname))
-imagespath = ''
+learn_inf = load_learner(hf_hub_download("Okkoman/PokeFace151", modelname))
 
-# %% ../pokemonclassifier.ipynb 31
+# %% ../pokemonclassifier.ipynb 30
 import gradio as gr
 
 lang = 'en'
-title = "# PokeFace : What is this pokemon ?"
+title = "# PokeFace - What is this pokemon ?"
 description = "## An image classifier for 1st generation pokemons (001-151)"
 unknown = 'unknown'
 
@@ -32,9 +32,8 @@ prob_threshold = 0.75
 from flask import request
 if request:
     lang = request.headers.get("Accept-Language")
-    print(lang)
     if lang == 'fr':
-        title = "# PokeFace : Quel est ce pokemon ?"
+        title = "# PokeFace - Quel est ce pokemon ?"
         description = "## Un classifieur d'images pour les pokemons de 1ere génération (001-151)"
         unknown = 'inconnu'