Infinity-1995's picture
Update app.py
0593b67 verified
raw
history blame
1.35 kB
import streamlit as st
from transformers import pipeline
# --- App title and description ---
st.set_page_config(page_title="Fake Job / Lie Detector", layout="centered")
st.title("πŸ” Fake Job / Lie Detector")
st.write(
"Enter a job description below and the AI will predict if it's likely genuine or fake."
)
# --- Load zero-shot classification model ---
@st.cache_resource
def load_model():
return pipeline(
"zero-shot-classification",
model="typeform/distilbert-base-uncased-mnli"
)
classifier = load_model()
# --- Text input ---
job_description = st.text_area("Enter the job description here:")
# --- Button action ---
if st.button("Check Job"):
if not job_description.strip():
st.warning("⚠️ Please enter a job description first!")
else:
candidate_labels = ["genuine", "fake"]
result = classifier(job_description, candidate_labels)
label = result['labels'][0]
confidence = round(result['scores'][0]*100, 2)
# --- Display results with color ---
if label == "genuine":
st.success(f"βœ… Prediction: {label.upper()} ({confidence}%)")
else:
st.error(f"❌ Prediction: {label.upper()} ({confidence}%)")
# --- Footer ---
st.markdown("---")
st.markdown("Built with ❀️ using Hugging Face Transformers and Streamlit.")