JaphetHernandez commited on
Commit
094526c
·
verified ·
1 Parent(s): 131dd3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import pandas as pd
2
  from langchain.chains import LLMChain
3
- from langchain.llms import LLM
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 = LLM(modelo, tokenizer)
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):