jaydemirandilla commited on
Commit
228caa0
·
1 Parent(s): b08fdd4

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 = 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
 
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