chore: closes #36

2023-10-12 09:38:00 +01:00
parent 998a583bdb
commit c7c6cfcd00
4 changed files with 25 additions and 5 deletions
--- a/views/py/python_model_template.py
+++ b/views/py/python_model_template.py
@@ -27,6 +27,7 @@ DATA_DIR_PREPARE = DATA_DIR + "/"
 def pathToLabel(path):
  path = tf.strings.regex_replace(path, DATA_DIR_PREPARE, "")
  path = tf.strings.regex_replace(path, ".jpg", "")
+  path = tf.strings.regex_replace(path, ".jpeg", "")
  path = tf.strings.regex_replace(path, ".png", "")
  return table.lookup(tf.strings.as_string([path]))
  #return tf.strings.as_string([path])