Update app.py
Browse files
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","
|
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):
|