Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import joblib | |
| # Load model | |
| def load_model(): | |
| vectorizer = joblib.load('vectorizer.pkl') | |
| model = joblib.load('model.pkl') | |
| return vectorizer, model | |
| vectorizer, model = load_model() | |
| # UI | |
| st.title("Skill Extraction from Job Descriptions") | |
| job_text = st.text_area("Enter job description:", height=200) | |
| if st.button("Extract Skills"): | |
| if job_text: | |
| # Vectorize and predict | |
| text_vec = vectorizer.transform([job_text]) | |
| prediction = model.predict(text_vec) | |
| st.success("Extracted Skills:") | |
| st.write(prediction[0]) |