Upload app.py

app.py

@@ -36,6 +36,10 @@ def main():
             filename = "Random_Forest_trained_model.pkl"
             loaded_model = pickle.load(open(filename, 'rb'))
 
+        elif selected_model == "**Decision Tree**":
+            filename = "Decision_Tree_Regression_trained_model.pkl"
+            loaded_model = pickle.load(open(filename, 'rb'))
+
         elif selected_model == "**CatBoost**":
             filename = "CatBoostRegressor_trained_model.pkl"
             loaded_model = pickle.load(open(filename, 'rb'))