sensei-ml commited on
Commit
fc09e59
·
verified ·
1 Parent(s): 27e200d

Update model/model.py

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