Fix name error

app.py

@@ -66,6 +66,16 @@ dataset_mapping = {
 dataset_path = dataset_mapping.get(problem, "datasets/spam_detection.csv")
 df = pd.read_csv(dataset_path)
 
+
+# Model Initialization
+model_mapping = {
+    "KNN": KNeighborsClassifier() if task == "Classification" else KNeighborsRegressor(),
+    "SVM": SVC() if task == "Classification" else SVR(),
+    "Random Forest": RandomForestClassifier() if task == "Classification" else RandomForestRegressor(),
+    "Decision Tree": DecisionTreeClassifier() if task == "Classification" else DecisionTreeRegressor(),
+    "Perceptron": Perceptron()
+}
+
 # Generated AI Code
 generated_code = f"""
 # AI Model Code
@@ -149,15 +159,6 @@ if task == "Classification":
         smote = SMOTE()
         X_train, y_train = smote.fit_resample(X_train, y_train)
 
-# Model Initialization
-model_mapping = {
-    "KNN": KNeighborsClassifier() if task == "Classification" else KNeighborsRegressor(),
-    "SVM": SVC() if task == "Classification" else SVR(),
-    "Random Forest": RandomForestClassifier() if task == "Classification" else RandomForestRegressor(),
-    "Decision Tree": DecisionTreeClassifier() if task == "Classification" else DecisionTreeRegressor(),
-    "Perceptron": Perceptron()
-}
-
 model_instance = model_mapping[model]
 
 # Train Model