DeepCNNClassifier_Project/prediction_service/app.py

import streamlit as st
from PIL import Image
import numpy as np
import tensorflow as tf

"""
Deep Classifier
"""

model = tf.keras.models.load_model("model.h5")
uploaded_file = st.file_uploader("Choose a file")
if uploaded_file is not None:
    image = Image.open(uploaded_file)
    image = image.resize((224,224))
    img_array = np.array(image)
    image_array = np.expand_dims(image,axis=0)
    result = model.predict(img_array)
    argmax_index = np.argmax(result,axis=1)
    if argmax_index[0] == 0:
        st.image(image,caption="predicted: cat")
    else:
        st.image(image,caption="predicted: dog")