Update app.py

Update de seguridad :D

app.py

@@ -8,7 +8,7 @@ from pathlib import Path
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForSequenceClassification
 from unidecode import unidecode
 from datetime import datetime
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, login
 #***************************************************************************
 #Defining default paths for the model to work
 #***************************************************************************
@@ -18,8 +18,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 #***************************************************************************
 #Setting up variables
 #***************************************************************************
+# Token privado desde variable de entorno
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
-
+# Login con el token (opcional pero recomendable)
+login(token=HF_TOKEN)
 #***************************************************************************
 #Functions
 #***************************************************************************
@@ -288,16 +291,16 @@ if __name__ == '__main__':
 
     # Loading Saved Models  
     # Modelo Contexto
-    context_model = AutoModelForSequenceClassification.from_pretrained("tecuhtli/mori-context-model")    
-    cont_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-context-model")
+    context_model = AutoModelForSequenceClassification.from_pretrained("tecuhtli/mori-context-model", use_auth_token=HF_TOKEN)    
+    cont_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-context-model", use_auth_token=HF_TOKEN) )
     
     # Modelo Técnico
-    tec_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-tecnico-model")
-    tec_model = AutoModelForSeq2SeqLM.from_pretrained("tecuhtli/mori-tecnico-model")
+    tec_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-tecnico-model", use_auth_token=HF_TOKEN) 
+    tec_model = AutoModelForSeq2SeqLM.from_pretrained("tecuhtli/mori-tecnico-model", use_auth_token=HF_TOKEN) 
 
     # Modelo Social
-    soc_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-social-model")
-    soc_model = AutoModelForSeq2SeqLM.from_pretrained("tecuhtli/mori-social-model")
+    soc_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-social-model", use_auth_token=HF_TOKEN) 
+    soc_model = AutoModelForSeq2SeqLM.from_pretrained("tecuhtli/mori-social-model", use_auth_token=HF_TOKEN) 
 
     # Available Device
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")