thenerdface11 commited on
Commit
6dbb430
·
verified ·
1 Parent(s): 65966ae

Fix config.json model_type

Browse files
Files changed (1) hide show
  1. config.json +7 -12
config.json CHANGED
@@ -1,12 +1,7 @@
1
- {
2
- "model_type": "food_recognition",
3
- "architecture": "efficientnet_yolo",
4
- "num_classes": 101,
5
- "image_size": [224, 224],
6
- "confidence_threshold": 0.5,
7
- "max_detections": 10,
8
- "nutritional_database": true,
9
- "version": "1.0.0",
10
- "framework": "tensorflow",
11
- "task": "image-classification"
12
- }
 
1
+ {
2
+ "model_type": "efficientnet",
3
+ "architecture": "efficientnet_b0",
4
+ "num_classes": 101,
5
+ "image_size": [224, 224],
6
+ "task": "image-classification"
7
+ }