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"
|