sohomghosh commited on
Commit
e02a8d2
β€’
1 Parent(s): e0b9b06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,8 +1,9 @@
1
- import pickle
 
2
  import gradio as gr
3
  from sentence_transformers import SentenceTransformer
4
  import lightgbm
5
- lr_clf_finbert = pickle.load(open("clf_lgbm_finbert.pickle", 'rb'))
6
  model = SentenceTransformer('ProsusAI/finbert')
7
 
8
  def get_readability(text):
1
+ #import pickle
2
+ import joblib
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
  import lightgbm
6
+ lr_clf_finbert = joblib.load('lgb.pkl')#pickle.load(open("clf_lgbm_finbert.pickle", 'rb'))
7
  model = SentenceTransformer('ProsusAI/finbert')
8
 
9
  def get_readability(text):