Update app.py

app.py

@@ -20,19 +20,18 @@ def display_images(percentage, complexity):
     return raw_image, embeddings_image
 
 # Define the Gradio interface
-# Use sliders to make the user interaction smoother
-data_percentage_range = (10, 100)  # Allow percentages between 10 and 100
-task_complexity_range = (16, 32)   # Allow complexity between 16 and 32
+data_percentage_options = [10, 30, 50, 70, 100]  # Specific percentage values
+task_complexity_options = [16, 32]               # Specific task complexity values
 
 # Define the layout and appearance of the UI
 with gr.Blocks() as demo:
     gr.Markdown("# Raw vs. Embeddings Inference Results")
     gr.Markdown("Select a data percentage and task complexity to view the corresponding inference result for raw channels and embeddings.")
     
-    # Inputs (using sliders for smoother tuning)
+    # Inputs (using Dropdowns for discrete values)
     with gr.Column():
-        percentage_slider = gr.Slider(minimum=10, maximum=100, step=10, label="Percentage of Data for Training", value=10)
-        complexity_slider = gr.Slider(minimum=16, maximum=32, step=4, label="Task Complexity", value=16)
+        percentage_dropdown = gr.Dropdown(choices=data_percentage_options, label="Percentage of Data for Training", value=10)
+        complexity_dropdown = gr.Dropdown(choices=task_complexity_options, label="Task Complexity", value=16)
     
     # Outputs (display the images side by side and set a smaller size for the images)
     with gr.Row():
@@ -40,8 +39,8 @@ with gr.Blocks() as demo:
         embeddings_img = gr.Image(label="Embeddings", type="pil", width=300, height=300, interactive=False)  # Smaller image size
     
     # Trigger image updates when inputs change
-    percentage_slider.change(fn=display_images, inputs=[percentage_slider, complexity_slider], outputs=[raw_img, embeddings_img])
-    complexity_slider.change(fn=display_images, inputs=[percentage_slider, complexity_slider], outputs=[raw_img, embeddings_img])
+    percentage_dropdown.change(fn=display_images, inputs=[percentage_dropdown, complexity_dropdown], outputs=[raw_img, embeddings_img])
+    complexity_dropdown.change(fn=display_images, inputs=[percentage_dropdown, complexity_dropdown], outputs=[raw_img, embeddings_img])
 
 # Launch the app
 if __name__ == "__main__":