Spaces:
Build error
Build error
sohomghosh
commited on
Commit
β’
b6ce5cc
1
Parent(s):
eab814a
Update app.py
Browse files
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):
|