App to convert a diagram drawing into a pytorch model code
to start app run:
streamlit run app.py
drawit.mp4
train model to convert sketch to clean diagram:
download didi dataset
run python3 preprocess.py
to prepare folders
run python3 train.py
to train model