manuelcozar55 commited on
Commit
e988252
verified
1 Parent(s): dbdb022

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -9,6 +9,9 @@ from pathlib import Path
9
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
10
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
11
  import torch
 
 
 
12
 
13
  # Configuraci贸n del modelo de resumen
14
  llm = HuggingFaceEndpoint(
@@ -17,6 +20,7 @@ llm = HuggingFaceEndpoint(
17
  max_new_tokens=4096,
18
  temperature=0.5,
19
  do_sample=False,
 
20
  )
21
  llm_engine_hf = ChatHuggingFace(llm=llm)
22
 
 
9
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
10
  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
  max_new_tokens=4096,
21
  temperature=0.5,
22
  do_sample=False,
23
+ token=HUGGINGFACE_TOKEN
24
  )
25
  llm_engine_hf = ChatHuggingFace(llm=llm)
26