Waflon commited on
Commit
966e1c3
1 Parent(s): c686a73

Update modelo.py

Browse files
Files changed (1) hide show
  1. modelo.py +2 -1
modelo.py CHANGED
@@ -12,7 +12,8 @@ def get_chain():
12
  # agregada en la config de hugginface
13
  #Embeddings que transforman a vectores densos multidimensionales las preguntas del SII
14
  embeddings = HuggingFaceEmbeddings(
15
- model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", # Ruta a modelo Pre entrenado
 
16
  model_kwargs={'device':'cpu'}, # Opciones de configuracion del modelo
17
  encode_kwargs={'normalize_embeddings': False} # Opciones de Encoding
18
  )
 
12
  # agregada en la config de hugginface
13
  #Embeddings que transforman a vectores densos multidimensionales las preguntas del SII
14
  embeddings = HuggingFaceEmbeddings(
15
+ model_name="sentence-transformers/multi-qa-mpnet-base-dot-v1", # Modelo correcto para embeddings que serán usados como QA
16
+ #model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", # Ruta a modelo Pre entrenado
17
  model_kwargs={'device':'cpu'}, # Opciones de configuracion del modelo
18
  encode_kwargs={'normalize_embeddings': False} # Opciones de Encoding
19
  )