wismaeka commited on
Commit
06d9739
1 Parent(s): 199f690

Update prediction.py

Browse files
Files changed (1) hide show
  1. prediction.py +1 -1
prediction.py CHANGED
@@ -4,7 +4,7 @@ import streamlit as st
4
 
5
  @st.cache_resource
6
  def load_model():
7
- with open('model_svm.pkl', 'rb') as file:
8
  model = pickle.load(file)
9
  return model
10
 
 
4
 
5
  @st.cache_resource
6
  def load_model():
7
+ with open('model_rf.pkl', 'rb') as file:
8
  model = pickle.load(file)
9
  return model
10