using flask to run pytorch model - View it on GitHub