AlignAI commited on
Commit
52ee0e5
·
verified ·
1 Parent(s): c0a4a7d

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -3
src/streamlit_app.py CHANGED
@@ -53,13 +53,13 @@ st.markdown("""
53
  def load_model():
54
  try:
55
  # Try loading the smart pipeline first
56
- model = joblib.load('svm_pipeline_model.pkl')
57
  return model, None
58
  except:
59
  try:
60
  # Fallback to separate files
61
- model = joblib.load('svm_model.pkl')
62
- scaler = joblib.load('scaler.pkl')
63
  return model, scaler
64
  except FileNotFoundError:
65
  return None, None
 
53
  def load_model():
54
  try:
55
  # Try loading the smart pipeline first
56
+ model = joblib.load('src/svm_model.pkl')
57
  return model, None
58
  except:
59
  try:
60
  # Fallback to separate files
61
+ model = joblib.load('src/svm_model.pkl')
62
+ scaler = joblib.load('src/scaler.pkl')
63
  return model, scaler
64
  except FileNotFoundError:
65
  return None, None