Chris4K commited on
Commit
cf6e4d7
·
verified ·
1 Parent(s): 4a2f2aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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 = "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,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
  )