File size: 1,478 Bytes
e7a389c
 
 
 
 
 
82b31d8
 
e7a389c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c852ee4
e7a389c
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import joblib
import gradio as gr
import numpy as np

def predict_employability(manner_of_speaking, self_confidence, ability_to_present_ideas, communication_skills, mental_alertness):
    # Load the updated model and label encoder
    model = joblib.load("employability.joblib")
    label_encoder = joblib.load("label_encoder.joblib") 
    
    # Prepare the input data
    input_data = np.array([[manner_of_speaking, self_confidence, ability_to_present_ideas, communication_skills, mental_alertness]])
    
    # Make prediction
    prediction = model.predict(input_data)[0]
    
    # Decode the prediction
    result = label_encoder.inverse_transform([prediction])[0]
    
    # Return the result with an emoji
    if result == "Employable":
        return f"✅ {result}"
    else:
        return f"😞 {result}"

# Define the Gradio interface
iface = gr.Interface(
    fn=predict_employability,
    inputs=[
        gr.Slider(1, 5, step=1, label="Manner of Speaking"),
        gr.Slider(1, 5, step=1, label="Self-Confidence"),
        gr.Slider(1, 5, step=1, label="Ability to Present Ideas"),
        gr.Slider(1, 5, step=1, label="Communication Skills"),
        gr.Slider(1, 5, step=1, label="Mental Alertness")
    ],
    outputs=gr.Textbox(label="Prediction"),
    title="Employability Prediction",
    description="Rate yourself on the given attributes (1-5) to check your employability status."
)

# Run the Gradio app
if __name__ == "__main__":
    iface.launch()