Raphaël Bournhonesque commited on
Commit
a332564
·
1 Parent(s): 3dc0018

Add new model

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -9,7 +9,8 @@ import streamlit as st
9
 
10
  MODEL_DESCRIPTION = """
11
  **keras_2_0**: Current production model\n
12
- **keras_sota_3_0**: Best model (SOTA), trained on 300 epochs with product name, ingredients, OCR-extracted ingredients and nutriments as input\n
 
13
  **keras_ingredient_ocr_3_0**: same as `keras_sota_3_0`, but trained on less epochs\n
14
  **keras_baseline_3_0**: model trained with product name, ingredients and nutriments as input\n
15
  **keras_original_3_0**: same inputs as **keras_2_0** (product name + ingredients), but retrained\n
@@ -21,7 +22,8 @@ http_session = requests.Session()
21
  @enum.unique
22
  class NeuralCategoryClassifierModel(enum.Enum):
23
  keras_2_0 = "keras-2.0"
24
- keras_sota_3_0 = "keras-sota-3-0"
 
25
  keras_ingredient_ocr_3_0 = "keras-ingredient-ocr-3.0"
26
  keras_baseline_3_0 = "keras-baseline-3.0"
27
  keras_original_3_0 = "keras-original-3.0"
 
9
 
10
  MODEL_DESCRIPTION = """
11
  **keras_2_0**: Current production model\n
12
+ **keras_image_embeddings_3_0**: same as `keras_300_epochs_3_0` but with image embedding as input\n
13
+ **keras_300_epochs_3_0**: trained on 300 epochs with product name, ingredients, OCR-extracted ingredients and nutriments as input\n
14
  **keras_ingredient_ocr_3_0**: same as `keras_sota_3_0`, but trained on less epochs\n
15
  **keras_baseline_3_0**: model trained with product name, ingredients and nutriments as input\n
16
  **keras_original_3_0**: same inputs as **keras_2_0** (product name + ingredients), but retrained\n
 
22
  @enum.unique
23
  class NeuralCategoryClassifierModel(enum.Enum):
24
  keras_2_0 = "keras-2.0"
25
+ keras_image_embeddings_3_0 = "keras-image-embeddings-3-0"
26
+ keras_300_epochs_3_0 = "keras-300-epochs-3-0"
27
  keras_ingredient_ocr_3_0 = "keras-ingredient-ocr-3.0"
28
  keras_baseline_3_0 = "keras-baseline-3.0"
29
  keras_original_3_0 = "keras-original-3.0"