friesti1 commited on
Commit
e7b7c5d
1 Parent(s): 5a15b4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -9,13 +9,12 @@ def predict_biome(uploaded_file):
9
  if uploaded_file is None:
10
  return "No file uploaded.", None, "No prediction"
11
 
12
- model = tf.keras.models.load_model('biomes-xception-model.keras')
13
 
14
  # Load the image from the file path
15
- with Image.open(uploaded_file).convert('RGB') as img:
16
  img = img.resize((150, 150))
17
  img_array = np.array(img)
18
- img_array = (img_array / 127.5) - 1.0 # Rescale the image as the model expects
19
 
20
  prediction = model.predict(np.expand_dims(img_array, axis=0))
21
 
 
9
  if uploaded_file is None:
10
  return "No file uploaded.", None, "No prediction"
11
 
12
+ model = tf.keras.models.load_model('biomes-vgg-model.keras')
13
 
14
  # Load the image from the file path
15
+ with Image.open(uploaded_file).convert('RGB') as img:
16
  img = img.resize((150, 150))
17
  img_array = np.array(img)
 
18
 
19
  prediction = model.predict(np.expand_dims(img_array, axis=0))
20