bug fix

app.py

@@ -236,7 +236,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     
     print("Loading model")
     
-    model_loading_name = model_path + "%s_%i_planet_detection" % (model_name, num_channels)
+    model_loading_name = "%s_%i_planet_detection" % (model_name, num_channels)
     
     if 'eff' in model_name:
         hparams = effnet_hparams[num_channels]
@@ -270,7 +270,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     # model = EfficientNetPreTrained.from_pretrained(model_loading_name)
     # model = AutoModel.from_pretrained(model_loading_name, trust_remote_code=True)
     
-    model = AutoModelForImageClassification.from_pretrained(model_loading_name)
+    model = AutoModelForImageClassification.from_pretrained(model_path + model_loading_name)
     
     # model = EfficientNet(dropout=hparams.dropout,
     #                                 num_channels=hparams.num_channels,