JadAssaf commited on
Commit
96fb78a
·
1 Parent(s): 5a5a555

requirements

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -2,6 +2,8 @@
2
  import gradio as gr
3
  import pickle
4
  import imblearn
 
 
5
  file_name = 'TSPI_model.sav'
6
  model = pickle.load(open(file_name,'rb'))
7
 
 
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
 
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  gradio==3.15.0
2
  imbalanced_learn==0.9.1
3
  imblearn==0.0
4
- sklearn==0.0
 
 
1
  gradio==3.15.0
2
  imbalanced_learn==0.9.1
3
  imblearn==0.0
4
+ sklearn==0.0
5
+ xgboost==1.6.2