kgemera commited on
Commit
9b7003c
·
verified ·
1 Parent(s): 33eb9e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -4,8 +4,13 @@ import pandas as pd
4
  from huggingface_hub import hf_hub_download
5
  import joblib
6
 
7
- model_path = hf_hub_download("kgemera/Biomedic_Text_Classifier", "classification_model.pkl")
8
- model = joblib.load(model_path)
 
 
 
 
 
9
 
10
  # Cargar el modelo
11
  model = joblib.load("classification_model.pkl")
 
4
  from huggingface_hub import hf_hub_download
5
  import joblib
6
 
7
+ hf_token = os.environ["HF_TOKEN"]
8
+
9
+ file_path = hf_hub_download(
10
+ repo_id="kgemera/Biomedic_Text_Classifier",
11
+ filename="classification_model.pkl",
12
+ use_auth_token=hf_token
13
+ )
14
 
15
  # Cargar el modelo
16
  model = joblib.load("classification_model.pkl")