ElBeh commited on
Commit
67038d7
·
verified ·
1 Parent(s): da35ee1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ local_model_path = snapshot_download(repo_id="ElBeh/ma_basemodel")
9
  model = keras.models.load_model(local_model_path)
10
 
11
  def get_prediction(img):
12
- x = image.img_to_array(img)
13
  x = np.expand_dims(x, axis=0)
14
  predictions = model.predict(x)
15
  return predictions
 
9
  model = keras.models.load_model(local_model_path)
10
 
11
  def get_prediction(img):
12
+ x = np.array(img)
13
  x = np.expand_dims(x, axis=0)
14
  predictions = model.predict(x)
15
  return predictions