Spaces:
Sleeping
Sleeping
Raphaël Bournhonesque
commited on
Commit
·
a332564
1
Parent(s):
3dc0018
Add new model
Browse files
app.py
CHANGED
@@ -9,7 +9,8 @@ import streamlit as st
|
|
9 |
|
10 |
MODEL_DESCRIPTION = """
|
11 |
**keras_2_0**: Current production model\n
|
12 |
-
**
|
|
|
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 |
-
|
|
|
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"
|