ancerlop commited on
Commit
bcd061e
1 Parent(s): 883cb4f

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -50
app.py DELETED
@@ -1,50 +0,0 @@
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
-
17
- # Establecer el umbral de similitud (puedes ajustarlo seg煤n tus necesidades)
18
- umbral_similitud = -11
19
-
20
- # Funci贸n para la predicci贸n
21
- def predecir_similitud(texto):
22
- # Vectorizar el texto (usando la misma funci贸n que antes)
23
- texto_vectorizado = vectorize_text(texto)
24
-
25
- # Buscar similitud con Faiss
26
- texto_vectorizado_np = np.array([texto_vectorizado], dtype=np.float32)
27
- D, I = index.search(texto_vectorizado_np, 1) # Buscar el vector m谩s cercano
28
- similitud = 1 - D[0][0] # Calcular similitud de coseno
29
-
30
- # Comparar con el umbral y proporcionar el resultado
31
- if similitud >= umbral_similitud:
32
- resultado = f"Coincidencia detectada (similitud: {similitud:.2f})"
33
- else:
34
- resultado = f"No se detect贸 coincidencia (similitud: {similitud:.2f})"
35
-
36
- return resultado
37
-
38
- # Crear la interfaz de Gradio con un bot贸n de "Predecir"
39
- iface = gr.Interface(
40
- fn=predecir_similitud,
41
- inputs=gr.inputs.Textbox(default="Ingrese su texto aqu铆"),
42
- outputs=gr.outputs.Textbox(),
43
- live=False,
44
- title="Detector de Coincidencias",
45
- description="Ingrese un texto y haga clic en 'Predecir' para detectar coincidencias con los vectores."
46
- )
47
-
48
- # Ejecutar la interfaz
49
- iface.launch()
50
-