Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,11 @@ from tensorflow.keras.applications.inception_resnet_v2 import InceptionResNetV2
|
|
| 8 |
|
| 9 |
#os.environ['TF_KERAS'] = '1'
|
| 10 |
#os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
|
|
|
|
|
|
|
|
|
|
| 11 |
model=load_model('best_model.h5')
|
|
|
|
| 12 |
|
| 13 |
|
| 14 |
def classify_image(inp):
|
|
|
|
| 8 |
|
| 9 |
#os.environ['TF_KERAS'] = '1'
|
| 10 |
#os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
|
| 11 |
+
|
| 12 |
+
labels=['Burger King', 'Subway', 'McDonalds', 'Starbucks', 'Other', 'KFC']
|
| 13 |
+
HEIGHT,WIDTH=224,224
|
| 14 |
model=load_model('best_model.h5')
|
| 15 |
+
NUM_CLASSES=6
|
| 16 |
|
| 17 |
|
| 18 |
def classify_image(inp):
|