sohomghosh commited on
Commit
b6ce5cc
β€’
1 Parent(s): eab814a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,7 +3,9 @@ import pickle
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
  import lightgbm
6
- lr_clf_finbert = pickle.load(open("lr_clf_finbert.pickle", 'rb')) #joblib.load('lgb.pkl')
 
 
7
  model = SentenceTransformer('ProsusAI/finbert')
8
 
9
  def get_readability(text):
 
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
  import lightgbm
6
+ lr_clf_finbert = pickle.load(open("lr_clf_finread_new.pkl",'rb')) #pickle.load(open("lr_clf_finbert.pickle", 'rb')) #joblib.load('lgb.pkl')
7
+ #lr_clf_finread_new.pkl is a Logistic Regression model trained on 13K intances
8
+ #lr_clf_finbert.pickle is a Logistic Regression model trained on 8K instances
9
  model = SentenceTransformer('ProsusAI/finbert')
10
 
11
  def get_readability(text):