drkareemkamal's picture
Create app.py
27eede3 verified
raw
history blame contribute delete
360 Bytes
import os
import numpy as np
import cv2
import tensorflow as tf
import streamlit as st
from tensorflow.keras.models import load_model
model = "AROI_image_segmentation.keras"
# Load the model
model = load_model(model)
print(f"The model loaded successfully")
model.compile(
loss = 'categorical_crossentropy',
optimizer = tf.keras.optimizers.Adam(lr))