Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ embed_model = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-
|
|
38 |
|
39 |
# Crea un objeto Chroma con el nombre de la colección
|
40 |
vectorstore = Chroma(
|
41 |
-
collection_name="
|
42 |
embedding_function=embed_model,
|
43 |
)
|
44 |
|
@@ -49,17 +49,27 @@ vectorstore.add_texts(context_data)
|
|
49 |
retriever = vectorstore.as_retriever()
|
50 |
|
51 |
# Crea un objeto PromptTemplate con el prompt
|
52 |
-
template = ("""
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
y
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
|
64 |
Context: {context}
|
65 |
Question: {question}
|
@@ -94,12 +104,12 @@ def rag_memory_stream(message, history):
|
|
94 |
|
95 |
# Crea un objeto Gradio con la función y el título
|
96 |
examples = [
|
97 |
-
"¿
|
98 |
-
"¿
|
99 |
-
"¿
|
100 |
]
|
101 |
-
description = "
|
102 |
-
title = "Experto en
|
103 |
demo = gr.ChatInterface(fn=rag_memory_stream,
|
104 |
type="messages",
|
105 |
title=title,
|
|
|
38 |
|
39 |
# Crea un objeto Chroma con el nombre de la colección
|
40 |
vectorstore = Chroma(
|
41 |
+
collection_name="unipapel_papeleria",
|
42 |
embedding_function=embed_model,
|
43 |
)
|
44 |
|
|
|
49 |
retriever = vectorstore.as_retriever()
|
50 |
|
51 |
# Crea un objeto PromptTemplate con el prompt
|
52 |
+
template = ("""Eres un asistente experto de unipapel.es, especializado en temas relacionados con la papelería. Tu función es ofrecer orientación a los clientes que realicen preguntas. Si no encuentras una respuesta, invítalos a contactar con nosotros. Puedes proporcionar información sobre nuestras categorías de productos según lo que te soliciten. La URL de la tienda es www.unipapel.es. A continuación, algunas de nuestras categorías:
|
53 |
+
|
54 |
+
Accesorios para pizarras
|
55 |
+
|
56 |
+
Acuarelas y lápices acuarelables
|
57 |
+
|
58 |
+
Archivadores de palanca
|
59 |
+
|
60 |
+
Blocs de dibujo
|
61 |
+
|
62 |
+
Bolígrafos
|
63 |
+
|
64 |
+
Carpetas clasificadoras
|
65 |
+
|
66 |
+
Cuadernos de caligrafía
|
67 |
+
|
68 |
+
Papelería escolar
|
69 |
+
|
70 |
+
Pegamentos y adhesivos
|
71 |
+
|
72 |
+
Pizarras blancas
|
73 |
|
74 |
Context: {context}
|
75 |
Question: {question}
|
|
|
104 |
|
105 |
# Crea un objeto Gradio con la función y el título
|
106 |
examples = [
|
107 |
+
"¿Qué tipo de bolígrafos son ideales para tomar apuntes en clase?",
|
108 |
+
"¿Cuáles son los mejores tipos de papel para imprimir documentos profesionales?",
|
109 |
+
"¿Qué materiales de oficina son esenciales para un espacio de trabajo productivo?"
|
110 |
]
|
111 |
+
description = "Chatbot con Unipapel"
|
112 |
+
title = "Experto en papeleria y todo lo referente"
|
113 |
demo = gr.ChatInterface(fn=rag_memory_stream,
|
114 |
type="messages",
|
115 |
title=title,
|