You can save your models as model1.py and model2.py with model names as my_model1 and my_model2 respectively then in main.py you can import these model as -
from model1 import my_model1
from model2 import my_model2
I guess this should work.