Update app.py

app.py

@@ -16,7 +16,7 @@ df = pd.DataFrame(dataset['train'])
 df.replace("No Data", pd.NA, inplace=True)
 
 # Handle missing numerical values with mean
-df.fillna(df.mean(), inplace=True)
+df.fillna(df.select_dtypes(include=['number']).mean(), inplace=True)
 
 # Handle missing categorical values with mode
 for col in df.select_dtypes(include=['object']).columns: