| import streamlit as st | |
| import sys | |
| import os | |
| sys.path.insert(1, os.path.join(sys.path[0], '..')) | |
| from utils import load_model, file_uploader_cb, load_ocr_model | |
| if __name__ == "__main__": | |
| # set page configurations and display/annotation options | |
| st.set_page_config( | |
| page_title="Circuit Sketch Recognizer", | |
| layout="wide" | |
| ) | |
| with st.sidebar: | |
| font_size = st.slider(label="Font Size", min_value=6, max_value=64, step=1, value=24) | |
| line_width = st.slider(label="Bounding Box Line Thickness", min_value=1, max_value=8, step=1, value=3) | |
| model = load_model() | |
| ocr_model, ocr_processor = load_ocr_model() | |
| # File uploader allows user to add their own image | |
| uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"]) | |
| if uploaded_file is not None: | |
| file_uploader_cb(model, ocr_model, ocr_processor, uploaded_file, font_size, line_width) | |