python src/neural_style.py train --dataset --style-image images/style_to_train/sketch1.jpg --save-model-dir models/ --epochs 2 --cuda 0
streamlit run src/main.py
Built an interactive deep learning app with Streamlit and PyTorch to apply for style transfer.
Python