Update app.py

app.py

@@ -15,8 +15,10 @@ model_name = "meta-llama/llama-3.1-8b-instruct"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
+pipe = pipeline("feature-extraction", model=model, tokenizer=tokenizer)
+
 # Usar transformers pipeline para cargar el modelo y tokenizer
-llm_pipeline = HuggingFacePipeline(model=model, tokenizer=tokenizer)
+llm_pipeline = HuggingFacePipeline(pipeline=pipe)
 
 # Interfaz de Streamlit
 st.title("Cosine Similarity con Llama 3.1")