Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ from pypdf import PdfReader
|
|
10 |
from sentence_transformers import SentenceTransformer
|
11 |
|
12 |
model_name = os.environ.get("MODEL", "Snowflake/snowflake-arctic-embed-m")
|
13 |
-
chunk_size = int(os.environ.get("CHUNK_SIZE",
|
14 |
-
default_max_characters = int(os.environ.get("DEFAULT_MAX_CHARACTERS",
|
15 |
|
16 |
model = SentenceTransformer(model_name)
|
17 |
# model.to(device="cuda")
|
@@ -132,6 +132,6 @@ gr.Interface(
|
|
132 |
gr.Number(label="Max output characters", value=default_max_characters),
|
133 |
],
|
134 |
outputs=[gr.JSON(label="Relevant chunks")],
|
135 |
-
title="
|
136 |
-
description="
|
137 |
).launch()
|
|
|
10 |
from sentence_transformers import SentenceTransformer
|
11 |
|
12 |
model_name = os.environ.get("MODEL", "Snowflake/snowflake-arctic-embed-m")
|
13 |
+
chunk_size = int(os.environ.get("CHUNK_SIZE", 128))
|
14 |
+
default_max_characters = int(os.environ.get("DEFAULT_MAX_CHARACTERS", 258))
|
15 |
|
16 |
model = SentenceTransformer(model_name)
|
17 |
# model.to(device="cuda")
|
|
|
132 |
gr.Number(label="Max output characters", value=default_max_characters),
|
133 |
],
|
134 |
outputs=[gr.JSON(label="Relevant chunks")],
|
135 |
+
title="",
|
136 |
+
description="",
|
137 |
).launch()
|