Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,9 @@ def predict_income(features):
|
|
16 |
# Use the Hugging Face model to predict the income
|
17 |
prediction = model(input_text)[0]
|
18 |
|
|
|
|
|
|
|
19 |
# Return the predicted income
|
20 |
return prediction['label']
|
21 |
|
|
|
16 |
# Use the Hugging Face model to predict the income
|
17 |
prediction = model(input_text)[0]
|
18 |
|
19 |
+
# Print the prediction for debugging
|
20 |
+
print("Prediction:", prediction)
|
21 |
+
|
22 |
# Return the predicted income
|
23 |
return prediction['label']
|
24 |
|