tecuhtli commited on
Commit
ae30a52
·
verified ·
1 Parent(s): 7a12538

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,9 +21,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
  # Token privado desde variable de entorno
22
  HF_TOKEN = os.environ.get("HF_TOKEN")
23
 
24
- # Login con el token (opcional pero recomendable)
25
- login(token=HF_TOKEN)
26
-
27
  #***************************************************************************
28
  #Functions
29
  #***************************************************************************
 
21
  # Token privado desde variable de entorno
22
  HF_TOKEN = os.environ.get("HF_TOKEN")
23
 
24
+ model = AutoModelForSequenceClassification.from_pretrained(
25
+ "tecuhtli/mori-context-model",
26
+ use_auth_token=HF_TOKEN)
27
  #***************************************************************************
28
  #Functions
29
  #***************************************************************************