jaydemirandilla commited on
Commit
4cbee86
·
1 Parent(s): 45ca24a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,9 +8,9 @@ model = tf.keras.models.load_model('denseNet121.h5')
8
  def classify_food_vs_nonfood(image):
9
  try:
10
  # Preprocess image
11
- #image_size = (224, 224)
12
- image_np = np.array(image.resize((224, 224))) / 255.0
13
- #image_np = np.array(image) / 255.0
14
  image_np_expanded = np.expand_dims(image_np, axis=0)
15
 
16
  # Make prediction
 
8
  def classify_food_vs_nonfood(image):
9
  try:
10
  # Preprocess image
11
+ image_size = (224, 224)
12
+ image = image.resize(image_size)
13
+ image_np = np.array(image) / 255.0
14
  image_np_expanded = np.expand_dims(image_np, axis=0)
15
 
16
  # Make prediction