patrickterrematte commited on
Commit
128f66f
·
verified ·
1 Parent(s): 0da98ec

Sync App files

Browse files
Files changed (1) hide show
  1. drug_app.py +3 -1
drug_app.py CHANGED
@@ -1,7 +1,9 @@
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
+ unknown_types = sio.get_untrusted_types(file="./Model/drug_pipeline.skops")
5
+ print(unknown_types)
6
+ pipe = sio.load("./Model/drug_pipeline.skops", trusted=unknown_types)
7
 
8
 
9
  def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio):