Spaces:
Runtime error
Runtime error
Commit
·
5e20048
1
Parent(s):
9e61d63
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ load_model = tf.keras.models.load_model('./resnet50_food_model')
|
|
9 |
|
10 |
def predict(img):
|
11 |
img = img.reshape(-1,224,224,3)
|
12 |
-
prediction=
|
13 |
|
14 |
return {labels[i]: float(prediction[i]) for i in range(2)}
|
15 |
|
|
|
9 |
|
10 |
def predict(img):
|
11 |
img = img.reshape(-1,224,224,3)
|
12 |
+
prediction= load_model.predict(img)[0]
|
13 |
|
14 |
return {labels[i]: float(prediction[i]) for i in range(2)}
|
15 |
|