JadAssaf
commited on
Commit
·
96fb78a
1
Parent(s):
5a5a555
requirements
Browse files- app.py +2 -0
- 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
|