Update app.py

app.py

@@ -16,7 +16,7 @@ RAW_PATH = os.path.join("images", "raw")
 EMBEDDINGS_PATH = os.path.join("images", "embeddings")
 
 # Specific values for percentage of data for training
-percentage_values = np.arange(9) + 1
+percentage_values = (np.arange(9) + 1)*100
 
 # Custom class to capture print output
 class PrintCapture(io.StringIO):
@@ -139,7 +139,7 @@ def identical_train_test_split(output_emb, output_raw, labels, percentage_idx):
     indices = torch.randperm(N)  # Randomly shuffle the indices
     
     # Calculate the split index
-    split_index = int(N * percentage_values[percentage_idx-1]/10)
+    split_index = int(N * percentage_values[percentage_idx]/100)
     print(f'Training Size: {split_index}')
     
     # Split indices into train and test
@@ -221,7 +221,7 @@ def process_hdf5_file(uploaded_file, percentage_idx):
         print(f"Output Raw Shape: {output_raw.shape}")
 
         print(f'percentage_idx: {percentage_idx}')
-        print(f'percentage_value: {percentage_values[percentage_idx-1]*10}')
+        print(f'percentage_value: {percentage_values[percentage_idx]}')
         train_data_emb, test_data_emb, train_data_raw, test_data_raw, train_labels, test_labels = identical_train_test_split(output_emb.view(len(output_emb),-1),
                                                                                                                              output_raw.view(len(output_raw),-1),
                                                                                                                              labels,
@@ -299,8 +299,8 @@ with gr.Blocks(css="""
             with gr.Column(elem_id="slider-container"):
                 gr.Markdown("Percentage of Data for Training")
                 #percentage_slider_los = gr.Slider(minimum=0, maximum=4, step=1, value=0, interactive=True, elem_id="vertical-slider")
-                percentage_dropdown_los = gr.Dropdown(choices=[str(v) for v in percentage_values*10], 
-                                                      value=10, 
+                percentage_dropdown_los = gr.Dropdown(choices=[0, 1, 2, 3, 4, 5, 6, 7, 8] 
+                                                      value=5, 
                                                       label="Percentage of Data for Training", 
                                                       interactive=True)