Update predict

predict

@@ -2,7 +2,7 @@
 !pip install transformers
 import torch
 from transformers import pipeline
-model_name = "model_name"  # Replace with the name or path of the pre-trained model
+model_name = skin_lesion_detection_model.h5  # Replace with the name or path of the pre-trained model
 classifier = pipeline("text-classification", model=model_name)
 
 text = "Skin lesion description"  # Replace with the skin lesion description you want to predict