keshavsingh2003 commited on
Commit
dc1ed39
·
verified ·
1 Parent(s): 583ad6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,12 +6,13 @@ import numpy as np
6
  from PIL import Image
7
 
8
  model_path = hf_hub_download(
9
- repo_id="keshavsingh2003/Clean-Brain-Tumor-Model",
10
- filename="clean_model.h5",
11
  repo_type="model"
12
  )
13
  model = load_model(model_path)
14
 
 
15
  class_labels = ['pituitary', 'glioma', 'notumor', 'meningioma']
16
 
17
  def predict_image(img):
 
6
  from PIL import Image
7
 
8
  model_path = hf_hub_download(
9
+ repo_id="keshavsingh2003/Brain-Tumor-Model-Keras",
10
+ filename="clean_model.keras",
11
  repo_type="model"
12
  )
13
  model = load_model(model_path)
14
 
15
+
16
  class_labels = ['pituitary', 'glioma', 'notumor', 'meningioma']
17
 
18
  def predict_image(img):