ThesisTest / app.py
NotRev's picture
Upload 4 files
df8e263 verified
raw
history blame contribute delete
636 Bytes
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])