NeerajCodz commited on
Commit
766afd5
·
1 Parent(s): 1309e50

Fixed prefix fraud_

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -505,8 +505,9 @@ async def llm_analyse(payload: LLMAnalysePayload):
505
  # Convert to DataFrame
506
  df = pd.DataFrame(transactions)
507
 
508
- # Remove 'fraud_' from all column names
509
- df.columns = [col.replace('fraud_', '') for col in df.columns]
 
510
 
511
  # Convert 'score' (previously 'fraud_score') to percentage string if it exists
512
  if 'score' in df.columns:
 
505
  # Convert to DataFrame
506
  df = pd.DataFrame(transactions)
507
 
508
+ # Remove 'fraud_' from all merchant names
509
+ if 'merchant' in df.columns:
510
+ df['merchant'] = df['merchant'].str.replace('fraud_', '', regex=False)
511
 
512
  # Convert 'score' (previously 'fraud_score') to percentage string if it exists
513
  if 'score' in df.columns: