Update app.py

app.py

@@ -28,6 +28,12 @@ def visualize_data(df):
     report = df.profile_report()
     spp(report)
 
+def reg_class(df,target):
+    if df[target].nunique() > 6:
+        return "regression"
+    else :
+        return "classification"
+
 def model_info(model,type_of_model):
     st.markdown("## Your Model Report")
     st.subheader("We've tried different models for you with its efficiency determining metrics")
@@ -452,9 +458,9 @@ if data is not None:
         if select_ == 'Visualize':
             visualize_data(df)
         if select_ == 'Preprocess':
-            type_ = preprocess_data(df,target)
+            preprocess_data(df,target)
         if select_ == 'Model Report':
-            model_info(reg,type_)
+            model_info(reg,reg_class(df,target))