updated

app.py

@@ -25,9 +25,8 @@ hp['dropout_rate'] = 0.1
 hp['class_names'] = ["breast_benign", "breast_malignant"]
 
 model = CNN_ViT(hp)
-model.load_weights("model/ResViT_for_breast_cancer_classification(1).keras")
-#model = tf.keras.models.load_model("model/resnet_for_breast_cancer-v1.h5")
-#model = from_pretrained_keras("ErnestBeckham/BreastResViT-II")
+model.load_weights("model/ResViT_for_breast_cancer_classification.keras")
+print("Model initiated")
 explainer = lime_image.LimeImageExplainer()