Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -111,9 +111,9 @@ def upload_file(file, model_name, split_strategy, chunk_size, overlap_size, max_
|
|
| 111 |
# Process files and get embeddings
|
| 112 |
|
| 113 |
if overlap_size is None:
|
| 114 |
-
overlap_size =
|
| 115 |
if chunk_size is None:
|
| 116 |
-
chunk_size =
|
| 117 |
|
| 118 |
embeddings, chunks = process_files(model_name, split_strategy, chunk_size, overlap_size, max_tokens)
|
| 119 |
|
|
@@ -134,10 +134,10 @@ iface = gr.Interface(
|
|
| 134 |
gr.Textbox(label="Search Query"),
|
| 135 |
gr.Dropdown(choices=list(MODELS.keys()), label="Embedding Model"),
|
| 136 |
gr.Radio(choices=["sentence", "recursive"], label="Split Strategy"),
|
| 137 |
-
gr.Slider(100, 1000, step=100, value=500, label="Chunk Size"),
|
| 138 |
-
gr.Slider(0, 100, step=10, value=50, label="Overlap Size"),
|
| 139 |
-
gr.Slider(50, 500, step=50, value=200, label="Max Tokens"),
|
| 140 |
-
gr.Slider(1, 10, step=1, value=5, label="Top K")
|
| 141 |
],
|
| 142 |
outputs="json"
|
| 143 |
)
|
|
|
|
| 111 |
# Process files and get embeddings
|
| 112 |
|
| 113 |
if overlap_size is None:
|
| 114 |
+
overlap_size = 0 # Set a default overlap size
|
| 115 |
if chunk_size is None:
|
| 116 |
+
chunk_size = 100 # Set a default chunk size
|
| 117 |
|
| 118 |
embeddings, chunks = process_files(model_name, split_strategy, chunk_size, overlap_size, max_tokens)
|
| 119 |
|
|
|
|
| 134 |
gr.Textbox(label="Search Query"),
|
| 135 |
gr.Dropdown(choices=list(MODELS.keys()), label="Embedding Model"),
|
| 136 |
gr.Radio(choices=["sentence", "recursive"], label="Split Strategy"),
|
| 137 |
+
gr.Slider(100, 1000, step=100, value=500, label="Chunk Size", type='int'), # Ensure type is int
|
| 138 |
+
gr.Slider(0, 100, step=10, value=50, label="Overlap Size", type='int'), # Ensure type is int
|
| 139 |
+
gr.Slider(50, 500, step=50, value=200, label="Max Tokens", type='int'), # Ensure type is int
|
| 140 |
+
gr.Slider(1, 10, step=1, value=5, label="Top K", type='int') # Ensure type is int
|
| 141 |
],
|
| 142 |
outputs="json"
|
| 143 |
)
|