Upload app.py

app.py

@@ -120,8 +120,6 @@ def input_layout():
 
 if __name__ == "__main__":
 
-    # Initialize image variable
-    poster = None
 
     ## --- Page config ------------ ##
     # Set page title
@@ -135,10 +133,15 @@ if __name__ == "__main__":
 
     # Set page footer
     footer()
+ 
+    ## --------------------------- ##
 
     # Load classification model
     model = CatBoostClassifier()      # parameters not required.
-    model.load_model('./train/model.cbm')
+    for root, dirs, files in os.walk("./"):
+        for file in files:
+            if file.endswith(".cbm"):
+                model.load_model(os.path.join(root, file))
 
     # Define inputs
     input_layout()