Hugging Face's logo

Spaces:
monaavr
/
Drug-Classification
Sleeping

App Files Community
monaavr commited on
Commit
92b0867
·
1 Parent(s): c84f32a

update trusted using get_trusted

Browse files
Files changed (1) hide show
  1. App/drug_app.py +1 -1
App/drug_app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import skops.io as sio
3
 
4
- pipe = sio.load("./Model/drug_pipeline.skops", trusted=True)
5
 
6
 
7
  def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio):
 
1
  import gradio as gr
2
  import skops.io as sio
3
 
4
+ sio.load("Model/drug_pipeline.skops", trusted=sio.get_untrusted_types(file = "Model/drug_pipeline.skops"))
5
 
6
 
7
  def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio):