sensei-ml commited on
Commit
e0801ac
·
verified ·
1 Parent(s): 7c2192b

Update model/model.py

Browse files
Files changed (1) hide show
  1. model/model.py +4 -4
model/model.py CHANGED
@@ -1,12 +1,12 @@
1
  from huggingface_hub import from_pretrained_keras
2
  from PIL import Image
 
3
  import numpy as np
 
4
  import tensorflow as tf
5
 
6
- import os
7
- os.environ["KERAS_BACKEND"] = "tensorflow"
8
- import keras
9
- model = keras.saving.load_model("hf://sensei-ml/Brain_Tumors_Classificator_CNN_Keras_Model")
10
 
11
  def predict(image_path):
12
  image = Image.fromarray(image_path)
 
1
  from huggingface_hub import from_pretrained_keras
2
  from PIL import Image
3
+ import keras
4
  import numpy as np
5
+ import os
6
  import tensorflow as tf
7
 
8
+ os.environ['KERAS_BACKEND'] = 'tensorflow'
9
+ model = keras.saving.load_model('hf://sensei-ml/Brain_Tumors_Classificator_CNN_Keras_Model')
 
 
10
 
11
  def predict(image_path):
12
  image = Image.fromarray(image_path)