ritampatra commited on
Commit
73e5aa0
·
verified ·
1 Parent(s): aaddfff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="juliensimon/xlm-roberta-base-finetuned-fraud-detection")
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")