Update app.py

app.py

@@ -28,8 +28,8 @@ def resize_image(image, input_shape=(224, 224, 3)):
 
 def load_model_file(filename):
     model_dir = snapshot_download(REPO_ID)
-    saved_model_dir = os.path.join(download_dir, filename)
-    unet_model = load_model(model_dir)
+    saved_model_filepath = os.path.join(model_dir, filename)
+    unet_model = load_model(saved_model_filepath)
     return unet_model
 
 def ensemble_predict(X_array):