Hugging Face's logo

Spaces:
azeemkhan417
/
BrainTumor
Sleeping

App Files Community
azeemkhan417 commited on
Commit
feff668
·
verified ·
1 Parent(s): fdbe415

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ class BrainTumorExplainer:
15
  Args:
16
  model_path (str): Path to the saved model weights
17
  """
18
- self.tumor_types = ["Glioma", "Pituitary Tumor","Meningioma"]
19
  self.model = self.load_model(model_path)
20
 
21
  def load_model(self, weights_path):
 
15
  Args:
16
  model_path (str): Path to the saved model weights
17
  """
18
+ self.tumor_types = ["Glioma","Meningioma","pituitary_tumor"]
19
  self.model = self.load_model(model_path)
20
 
21
  def load_model(self, weights_path):