JaphetHernandez commited on
Commit
611a5f2
·
verified ·
1 Parent(s): b19e79b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ if uploaded_file:
34
  sys.exit(1)
35
 
36
  # Crear una cadena LLM con LangChain usando HuggingFaceHub
37
- llm = HuggingFaceHub(repo_id=model_name, model_kwargs={"device_map": "auto"})
38
  chain = LLMChain(llm=llm)
39
 
40
  def calcular_similitud(texto):
 
34
  sys.exit(1)
35
 
36
  # Crear una cadena LLM con LangChain usando HuggingFaceHub
37
+ llm = HuggingFaceHub(repo_id=model_name, model_kwargs={"device_map": "auto"}, huggingfacehub_api_token=huggingface_token)
38
  chain = LLMChain(llm=llm)
39
 
40
  def calcular_similitud(texto):