manuelcozar55 commited on
Commit
cdd50f7
verified
1 Parent(s): 2cde75c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
11
  import torch
12
  import os
13
 
14
- HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
15
 
16
  # Configuraci贸n del modelo de resumen
17
  llm = HuggingFaceEndpoint(
@@ -20,7 +20,6 @@ llm = HuggingFaceEndpoint(
20
  max_new_tokens=4096,
21
  temperature=0.5,
22
  do_sample=False,
23
- model_kwargs={"token": HUGGINGFACE_TOKEN}
24
  )
25
  llm_engine_hf = ChatHuggingFace(llm=llm)
26
 
 
11
  import torch
12
  import os
13
 
14
+ login(token=os.getenv('HUGGINGFACE_TOKEN'))
15
 
16
  # Configuraci贸n del modelo de resumen
17
  llm = HuggingFaceEndpoint(
 
20
  max_new_tokens=4096,
21
  temperature=0.5,
22
  do_sample=False,
 
23
  )
24
  llm_engine_hf = ChatHuggingFace(llm=llm)
25