
Train 2 more models (resnet50 and resnet18) and then deploy them after adding this code to your own pycharm (make sure to change the model name and pkl names):
MODEL_PATHS = { "Cat vs Dog": "cat_vs_dog_model_fastai2_8_4.pkl", "Model 2": "model_2.pkl", "Model 3": "model_3.pkl", }
selected_model_name = st.selectbox( "Choose a model", list(MODEL_PATHS.keys()) ) model_path = MODEL_PATHS[selected_model_name] model = load_model(model_path)