Spaces:
Build error
Build error
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):
|