Spaces:

skavtech
/

Blood_Cancer_detection_model

No application file

skavtech commited on Jan 5

Commit

869580a

verified ·

1 Parent(s): e18e0c2

Application code

Files changed (1) hide show

model.py CHANGED Viewed

@@ -9,7 +9,7 @@ import gradio as gr
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 # Load the pre-trained model
-model = tf.keras.models.load_model("maheshbabu.h5")
 # Define class labels
 classes = ["Normal", "Cancerous"]

 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 # Load the pre-trained model
+model = tf.keras.models.load_model("blood_cancer_model.h5")
 # Define class labels
 classes = ["Normal", "Cancerous"]