dadn commited on
Commit
08757bc
1 Parent(s): 960faad

Sync App files

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