ancerlop commited on
Commit
c43be1e
·
1 Parent(s): a024273

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -1,7 +1,16 @@
 
1
  import gradio as gr
2
  import numpy as np
3
  import faiss
4
 
 
 
 
 
 
 
 
 
5
  # Cargar el índice de Faiss
6
  index = faiss.read_index("faiss_train_index.index")
7
 
 
1
+ from sentence_transformers import SentenceTransformer
2
  import gradio as gr
3
  import numpy as np
4
  import faiss
5
 
6
+ # Cargar el modelo de Sentence Transformers
7
+ model_name="quora-distilbert-multilingual"
8
+ model = SentenceTransformer(model_name)
9
+
10
+ # Función para vectorizar un texto
11
+ def vectorize_text(text):
12
+ return model.encode([text])[0]
13
+
14
  # Cargar el índice de Faiss
15
  index = faiss.read_index("faiss_train_index.index")
16