from train import train_model
from api import app


if __name__ == "__main__":
    choice = input("1 - Обучение модели; 2 - Запуск сервера\n")
    if choice == "1":
        train_model()
    else:
        app.run()