Spaces:
Sleeping
Sleeping
File size: 636 Bytes
df8e263 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import streamlit as st
import joblib
# Load model
@st.cache_resource
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]) |