Spaces:
Sleeping
Sleeping
Raphaël Bournhonesque
commited on
Commit
·
3dc0018
1
Parent(s):
4ebffee
improve app description
Browse files
app.py
CHANGED
@@ -7,6 +7,15 @@ import requests
|
|
7 |
import streamlit as st
|
8 |
|
9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
http_session = requests.Session()
|
11 |
|
12 |
@enum.unique
|
@@ -69,6 +78,8 @@ barcode = st.sidebar.text_input(
|
|
69 |
"Product barcode", default_barcode
|
70 |
)
|
71 |
threshold = st.sidebar.number_input("Threshold", format="%f", value=0.5) or None
|
|
|
|
|
72 |
model_names = st.multiselect(
|
73 |
"Name of the model",
|
74 |
[x.name for x in NeuralCategoryClassifierModel],
|
|
|
7 |
import streamlit as st
|
8 |
|
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
|
16 |
+
**keras_product_name_only_3_0**: model with only product name as input
|
17 |
+
"""
|
18 |
+
|
19 |
http_session = requests.Session()
|
20 |
|
21 |
@enum.unique
|
|
|
78 |
"Product barcode", default_barcode
|
79 |
)
|
80 |
threshold = st.sidebar.number_input("Threshold", format="%f", value=0.5) or None
|
81 |
+
|
82 |
+
st.sidebar.write("---\n# Model description\n" + MODEL_DESCRIPTION)
|
83 |
model_names = st.multiselect(
|
84 |
"Name of the model",
|
85 |
[x.name for x in NeuralCategoryClassifierModel],
|