Mahesh2309 commited on
Commit
55c83c5
·
verified ·
1 Parent(s): 2fb323a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
11
  model = tf.keras.models.load_model("maheshbabu.h5")
12
 
13
  # Define class labels
14
- classes = ["Normal","Early", "Pre-Cancerous","Post-Cancerous"]
15
 
16
  # Prediction function
17
  def predict(image):
 
11
  model = tf.keras.models.load_model("maheshbabu.h5")
12
 
13
  # Define class labels
14
+ classes = ["Normal","Cancerous"]
15
 
16
  # Prediction function
17
  def predict(image):