sensei-ml commited on
Commit
f13f395
1 Parent(s): 9eaaac7

Update model/model.py

Browse files
Files changed (1) hide show
  1. model/model.py +1 -1
model/model.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import os
5
  import tensorflow as tf
6
 
7
- model = keras.models.load_model("hf://sensei-ml/Brain_Tumors_Classificator_CNN_Keras_Model")
8
 
9
  def predict(image_path):
10
  image = Image.fromarray(image_path)
 
4
  import os
5
  import tensorflow as tf
6
 
7
+ model = keras.saving.load_model('model/model.keras')
8
 
9
  def predict(image_path):
10
  image = Image.fromarray(image_path)