ikigai-edu commited on
Commit
5e20048
·
1 Parent(s): 9e61d63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=model.predict(img)[0]
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