updating saving I suppose

app.py

@@ -220,13 +220,13 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
                                     depth_mult=hparams.depth_mult,
         )
     
-    config.save_pretrained(model_loading_name)
+    config.save_pretrained("%s_planet_detection" % (model_name))
     
-    config = EfficientNetConfig.from_pretrained(model_loading_name)
+    config = EfficientNetConfig.from_pretrained("%s_planet_detection" % (model_name))
     
     model = EfficientNetPreTrained(config)
 
-    pretrained_model = timm.create_model(model_loading_name, pretrained=True)
+    pretrained_model = timm.create_model("%s_planet_detection" % (model_name), pretrained=True)
     model.model.load_state_dict(pretrained_model.state_dict())
     
     # pipeline = pipeline(task="image-classification", model=model_loading_name)