eskayML commited on
Commit
72c609a
·
1 Parent(s): ac74472

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,7 +2,9 @@ import pickle
2
  import gradio as gr
3
 
4
 
5
- model = pickle.load(open('stress_1.0.pkl', 'rb'))
 
 
6
 
7
  def predict_stress(text):
8
  return model.predict([text])[0].upper()
 
2
  import gradio as gr
3
 
4
 
5
+ with open('stress_1.0.pkl','rb') as f:
6
+ model = pickle.load(f)
7
+ print(model.steps)
8
 
9
  def predict_stress(text):
10
  return model.predict([text])[0].upper()