Upload folder using huggingface_hub
Browse files
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 |
|