JadAssaf commited on
Commit
efb2a5c
·
1 Parent(s): 96fb78a
Files changed (3) hide show
  1. TSPI_model_joblib.sav +0 -0
  2. app.py +4 -2
  3. 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 = 'TSPI_model.sav'
8
- model = pickle.load(open(file_name,'rb'))
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