Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def predict_user_profile(inputs):
|
|
17 |
|
18 |
# Select the relevant feature columns used during model training
|
19 |
feature_columns_to_use = ['statuses_count', 'followers_count', 'friends_count',
|
20 |
-
'favourites_count',
|
21 |
df_features = df[feature_columns_to_use]
|
22 |
|
23 |
# Load the pre-trained model
|
|
|
17 |
|
18 |
# Select the relevant feature columns used during model training
|
19 |
feature_columns_to_use = ['statuses_count', 'followers_count', 'friends_count',
|
20 |
+
'favourites_count','listed_count'] # Assuming language isn't used
|
21 |
df_features = df[feature_columns_to_use]
|
22 |
|
23 |
# Load the pre-trained model
|