keikanih commited on
Commit
db6834c
Β·
verified Β·
1 Parent(s): 5a84672

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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", 500))
14
- default_max_characters = int(os.environ.get("DEFAULT_MAX_CHARACTERS", 500))
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="RAG Tool",
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()