Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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):
|