Spaces:
Running
Running
File size: 2,332 Bytes
548a54a |
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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import streamlit as st
import keras
import numpy as np
from PIL import Image
st.set_page_config(layout="wide")
#title
st.title('Crossing Identifier')
#header
st.header('Choose whether you\'d like to enter a latitude/longitude coordinates, or upload a satellite image.')
state = st.session_state
if "dict_options" not in state:
state.dict_options = {}
if "submitted" not in state:
state.submitted = False
options = ["option1", "option2", "option3", "option4"]
col1, col2, col3 = st.columns(3)
with col1.form("my_form"):
new_country = st.text_input("New country")
submit_button = st.form_submit_button(
label="Add new country", on_click=lambda: state.update(submitted=True)
)
if state.submitted:
state.dict_options[new_country] = col2.multiselect(
f"Select the options you want for {new_country}",
options,
default=options[:2],
)
col2.write(state.dict_options)
#divide app into two columns
col1, col2 = st.columns(2)
#load model and initialize image size required by model. uploaded images are resized to indicated size
loaded_model = keras.models.load_model("0.0008-0.92.keras")
img_height = 640
img_width = 640
#place to enter
#place to enter coordinates (or upload) and display image
with col1:
enter_coords = st.button("Enter Coordinates")
if enter_coords:
st.write(":smile:")
upload_img = st.button("Upload an Image")
if enter_coords:
st.write("ok")
st.header('Please upload a satellite image, or enter a latitude/longitude pair')
img_buffer = st.file_uploader("Upload a satellite image file (format: .png, .jpeg, or .jpg).",type=['png', 'jpeg', 'jpg'])
if img_buffer is not None:
st.image(img_buffer, use_column_width = True)
#place to display prediction result
with col2:
if img_buffer is not None:
st.header('Result')
img = Image.open(img_buffer).convert("RGB")
img_array = np.array(img)
batch_size = 1
img_array = np.reshape(img_array,[batch_size,img_height,img_width,3])
result = loaded_model.predict(img_array)
st.write("Your prediction is:")
st.write(f"{np.round(result[0][0]*100,decimals=2)}% chance of no crossing")
st.write(f"{np.round(result[0][1]*100,decimals=2)}% chance of at least one crossing")
|