File size: 1,399 Bytes
417de76 67d3079 417de76 b45c977 67d3079 417de76 67d3079 417de76 67d3079 417de76 67d3079 417de76 67d3079 417de76 67d3079 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import tensorflow as tf
from huggingface_hub import snapshot_download
from keras.preprocessing import image
import numpy as np
import gradio as gr
from keras.layers import TFSMLayer
# Download the model from Hugging Face Space to a local path
local_model_path = snapshot_download("syaha/skin_cancer_detection_model")
# Load the model using TFSMLayer
model = TFSMLayer(local_model_path, call_endpoint="serving_default")
# Class names for skin cancer classification (assuming you already know these)
class_names = ['akiec', 'bcc', 'bkl', 'df', 'nv', 'vasc', 'mel']
# Define a prediction function
def predict_skin_cancer(image_path):
# Preprocess the image for prediction
img = image.load_img(image_path, target_size=(224, 224)) # Resize to 224x224 (or use the correct input size)
img_array = image.img_to_array(img)
img_array = np.expand_dims(img_array, axis=0) / 255.0 # Normalize the image
# Make prediction
predictions = model.predict(img_array)
predicted_class = np.argmax(predictions, axis=1)[0] # Get the class index
predicted_label = class_names[predicted_class] # Map the index to the class name
return f"Predicted class: {predicted_label}"
# Set up Gradio interface to interact with the model
iface = gr.Interface(fn=predict_skin_cancer, inputs=gr.Image(type="filepath"), outputs="text", live=True)
# Launch the Gradio interface
iface.launch()
|