Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import pandas as pd
|
2 |
from langchain.chains import LLMChain
|
3 |
-
from langchain.llms import
|
4 |
from transformers import LLaMAForCausalLM, LLaMATokenizer
|
5 |
from huggingface_hub import login
|
6 |
import streamlit as st
|
@@ -19,7 +19,7 @@ modelo = LLaMAForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B")
|
|
19 |
tokenizer = LLaMATokenizer.from_pretrained("meta-llama/Llama-3.2-1B")
|
20 |
|
21 |
# Crear una cadena LLM con LangChain
|
22 |
-
llm =
|
23 |
chain = LLMChain(llm)
|
24 |
|
25 |
def calcular_similitud(texto):
|
|
|
1 |
import pandas as pd
|
2 |
from langchain.chains import LLMChain
|
3 |
+
from langchain.llms import HuggingFaceLLM
|
4 |
from transformers import LLaMAForCausalLM, LLaMATokenizer
|
5 |
from huggingface_hub import login
|
6 |
import streamlit as st
|
|
|
19 |
tokenizer = LLaMATokenizer.from_pretrained("meta-llama/Llama-3.2-1B")
|
20 |
|
21 |
# Crear una cadena LLM con LangChain
|
22 |
+
llm = HuggingFaceLLM(modelo, tokenizer)
|
23 |
chain = LLMChain(llm)
|
24 |
|
25 |
def calcular_similitud(texto):
|