tvachaAI / app.py
rajsecrets0's picture
Create app.py
31ec0da verified
raw history blame
No virus
564 Bytes
import streamlit as st
import tensorflow as tf
st.title("Skin Disease Classification App")
# Load model
model = tf.keras.models.load_model('TvachaAI.h5')
# Create sidebar for uploading images
st.sidebar.title("Upload Image")
image = st.sidebar.file_uploader("Upload an image", type=['png', 'jpg'])
if image is not None:
# Process & predict
img = process_image(image)
preds = model.predict(img)
# Display image & predictions
st.image(img, use_column_width=True)
st.write(f"Prediction: {decode_predictions(preds)}")