Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,11 +18,28 @@ trait_prefixes = {
|
|
18 |
'Openness': 'OPN'
|
19 |
}
|
20 |
|
21 |
-
#
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
|
27 |
# Inference function
|
28 |
def predict_personality(*inputs):
|
@@ -32,10 +49,7 @@ def predict_personality(*inputs):
|
|
32 |
personality = encoder.inverse_transform(pred)[0]
|
33 |
return f"Predicted Personality: **{personality}**"
|
34 |
|
35 |
-
# Gradio UI
|
36 |
-
inputs = [gr.Slider(1.0, 5.0, value=3.0, label=label) for label in feature_labels]
|
37 |
-
output = gr.Textbox(label="Prediction")
|
38 |
-
|
39 |
demo = gr.Interface(
|
40 |
fn=predict_personality,
|
41 |
inputs=inputs,
|
|
|
18 |
'Openness': 'OPN'
|
19 |
}
|
20 |
|
21 |
+
# Example IPIP-FFM questionnaire questions (replace with actual questions)
|
22 |
+
questions = [
|
23 |
+
"I am the life of the party.", "I feel comfortable around people.", "I often feel blue.", "I am easily disturbed.",
|
24 |
+
"I enjoy trying new things.", "I am talkative.", "I don't mind being the center of attention.", "I often get upset.",
|
25 |
+
"I am interested in abstract ideas.", "I am full of energy.", "I would rather sit at home than go out.",
|
26 |
+
"I don't like to draw attention to myself.", "I am sometimes easily angered.", "I have frequent mood swings.",
|
27 |
+
"I like to travel to new places.", "I like to meet new people.", "I enjoy having a wide variety of friends.",
|
28 |
+
"I am good at handling stress.", "I often feel like I’m not in control of my emotions.", "I prefer variety to routine.",
|
29 |
+
"I am a very anxious person.", "I prefer to stick to one activity at a time.", "I enjoy being active in social settings.",
|
30 |
+
"I don’t like to take risks.", "I get along with most people.", "I get bored easily.", "I tend to be impulsive.",
|
31 |
+
"I like being organized.", "I feel uncomfortable around strangers.", "I avoid conflict with others.", "I have a lot of energy.",
|
32 |
+
"I find it difficult to express my emotions.", "I often feel lonely.", "I like to keep my thoughts and feelings to myself.",
|
33 |
+
"I find it difficult to relax.", "I am always prepared.", "I sometimes feel down.", "I find it difficult to focus on one task.",
|
34 |
+
"I enjoy the company of others.", "I like to talk about my feelings.", "I can’t stand being interrupted.",
|
35 |
+
"I often forget to do things.", "I am good at understanding other people’s feelings.", "I enjoy taking on challenges.",
|
36 |
+
"I often feel overwhelmed.", "I like to take my time making decisions.", "I enjoy being in charge.", "I am easily distracted.",
|
37 |
+
"I get along well with others.", "I enjoy being the center of attention."
|
38 |
+
]
|
39 |
+
|
40 |
+
# Create sliders with questions
|
41 |
+
inputs = [gr.Slider(1.0, 5.0, value=3.0, label=q) for q in questions]
|
42 |
+
output = gr.Textbox(label="Prediction")
|
43 |
|
44 |
# Inference function
|
45 |
def predict_personality(*inputs):
|
|
|
49 |
personality = encoder.inverse_transform(pred)[0]
|
50 |
return f"Predicted Personality: **{personality}**"
|
51 |
|
52 |
+
# Gradio UI with 50 sliders and full question labels
|
|
|
|
|
|
|
53 |
demo = gr.Interface(
|
54 |
fn=predict_personality,
|
55 |
inputs=inputs,
|