JadAssaf
commited on
Commit
·
efb2a5c
1
Parent(s):
96fb78a
Update
Browse files- TSPI_model_joblib.sav +0 -0
- app.py +4 -2
- requirements.txt +3 -1
TSPI_model_joblib.sav
ADDED
Binary file (161 kB). View file
|
|
app.py
CHANGED
@@ -1,11 +1,13 @@
|
|
1 |
# %%
|
2 |
import gradio as gr
|
|
|
|
|
3 |
import pickle
|
4 |
import imblearn
|
5 |
import sklearn
|
6 |
import xgboost
|
7 |
-
file_name = '
|
8 |
-
model =
|
9 |
|
10 |
|
11 |
def STPI(t_1_MaxValue,t_2_MaxValue,t_2_0_MaxValue,
|
|
|
1 |
# %%
|
2 |
import gradio as gr
|
3 |
+
import joblib
|
4 |
+
import dill
|
5 |
import pickle
|
6 |
import imblearn
|
7 |
import sklearn
|
8 |
import xgboost
|
9 |
+
file_name = 'TSPI_model_joblib.sav'
|
10 |
+
model = joblib.load(file_name)
|
11 |
|
12 |
|
13 |
def STPI(t_1_MaxValue,t_2_MaxValue,t_2_0_MaxValue,
|
requirements.txt
CHANGED
@@ -2,4 +2,6 @@ gradio==3.15.0
|
|
2 |
imbalanced_learn==0.9.1
|
3 |
imblearn==0.0
|
4 |
sklearn==0.0
|
5 |
-
xgboost==1.6.2
|
|
|
|
|
|
2 |
imbalanced_learn==0.9.1
|
3 |
imblearn==0.0
|
4 |
sklearn==0.0
|
5 |
+
xgboost==1.6.2
|
6 |
+
joblib==1.2.0
|
7 |
+
dill==0.3.6
|