Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,12 +4,12 @@ from transformers import pipeline
|
|
4 |
# Load pretrained fraud detection model from Hugging Face
|
5 |
@st.cache_resource
|
6 |
def load_model():
|
7 |
-
return pipeline("text-classification", model="
|
8 |
|
9 |
model = load_model()
|
10 |
|
11 |
# Streamlit UI
|
12 |
-
st.title("💳 Fraud Detection System (Hugging Face Model)")
|
13 |
|
14 |
# User input text
|
15 |
user_input = st.text_area("Enter transaction description:", "Payment of $500 for electronics purchase")
|
|
|
4 |
# Load pretrained fraud detection model from Hugging Face
|
5 |
@st.cache_resource
|
6 |
def load_model():
|
7 |
+
return pipeline("text-classification", model="d4data/bert-base-cased-finetuned-fraud-detection")
|
8 |
|
9 |
model = load_model()
|
10 |
|
11 |
# Streamlit UI
|
12 |
+
st.title("💳 Financial Fraud Detection System (Hugging Face Model)")
|
13 |
|
14 |
# User input text
|
15 |
user_input = st.text_area("Enter transaction description:", "Payment of $500 for electronics purchase")
|