Update app.py

app.py

@@ -1,10 +1,3 @@
-# -*- coding: utf-8 -*-
-"""
-Created on Tue Sep 17 15:03:39 2024
-
-@author: salikha4
-"""
-
 import gradio as gr
 import os
 from PIL import Image
@@ -30,19 +23,25 @@ def display_images(percentage, complexity):
 data_percentage_options = [10, 30, 50, 70, 100]
 task_complexity_options = [16, 32]
 
-demo = gr.Interface(
-    fn=display_images,
-    inputs=[
-        gr.Dropdown(data_percentage_options, label="Percentage of Data for Training"),
-        gr.Radio(task_complexity_options, label="Task Complexity")
-    ],
-    outputs=[
-        gr.Image(label="Raw Channels"),
-        gr.Image(label="Embeddings")
-    ],
-    title="Raw vs. Embeddings Inference Results",
-    description="Select a data percentage and task complexity to view the corresponding inference result for raw channels and embeddings."
-)
+# 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 (data percentage and task complexity)
+    with gr.Row():
+        percentage_dropdown = gr.Dropdown(data_percentage_options, label="Percentage of Data for Training", value=10)
+        complexity_radio = gr.Radio(task_complexity_options, label="Task Complexity", value=16)
+    
+    # Outputs (display the images side by side)
+    with gr.Row():
+        raw_img = gr.Image(label="Raw Channels", type="pil", interactive=False)
+        embeddings_img = gr.Image(label="Embeddings", type="pil", interactive=False)
+    
+    # Update images instantaneously without clicking submit
+    percentage_dropdown.change(fn=display_images, inputs=[percentage_dropdown, complexity_radio], outputs=[raw_img, embeddings_img], live=True)
+    complexity_radio.change(fn=display_images, inputs=[percentage_dropdown, complexity_radio], outputs=[raw_img, embeddings_img], live=True)
 
+# Launch the app
 if __name__ == "__main__":
     demo.launch()