Update src/streamlit_app.py
Browse files- 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('
|
| 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
|