Spaces:
Build error
Build error
Update Image_Classification_EfficientNetLite4/app.py
Browse files- updated the code with the new directory of "labels_map.txt"
Image_Classification_EfficientNetLite4/app.py
CHANGED
@@ -10,7 +10,7 @@ from huggingface_hub import hf_hub_download
|
|
10 |
|
11 |
modele = hf_hub_download(repo_id="onnx/EfficientNet-Lite4", filename="efficientnet-lite4-11.onnx")
|
12 |
# load the labels text file
|
13 |
-
labels = json.load(open("
|
14 |
|
15 |
# set image file dimensions to 224x224 by resizing and cropping image from center
|
16 |
def pre_process_edgetpu(img, dims):
|
|
|
10 |
|
11 |
modele = hf_hub_download(repo_id="onnx/EfficientNet-Lite4", filename="efficientnet-lite4-11.onnx")
|
12 |
# load the labels text file
|
13 |
+
labels = json.load(open("Image_Classification_EfficientNetLite4/labels_map.txt", "r"))
|
14 |
|
15 |
# set image file dimensions to 224x224 by resizing and cropping image from center
|
16 |
def pre_process_edgetpu(img, dims):
|