papousek commited on
Commit
e581cad
·
verified ·
1 Parent(s): ed46054

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -77,7 +77,10 @@ import numpy as np
77
  import tensorflow as tf
78
 
79
  MODEL_PATH = hf_hub_download('kindwise/crossroad.tiny', 'model.tflite') # or model.optimized.tflite
 
80
 
 
 
81
  INTERPRETER = tf.lite.Interpreter(model_path=MODEL_PATH)
82
  INTERPRETER.allocate_tensors()
83
 
 
77
  import tensorflow as tf
78
 
79
  MODEL_PATH = hf_hub_download('kindwise/crossroad.tiny', 'model.tflite') # or model.optimized.tflite
80
+ CLASSES_PATH = hf_hub_download('kindwise/crossroad.tiny', 'classes.txt')
81
 
82
+ with open(CLASSES_PATH) as f:
83
+ CLASSES = [line.strip() for line in f]
84
  INTERPRETER = tf.lite.Interpreter(model_path=MODEL_PATH)
85
  INTERPRETER.allocate_tensors()
86