upload source of model file main.py

Update README.md

Added the model.pth

initial commit