Create app.py

gradio/app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+import os
+
+# Paths to the images folder (use relative paths)
+RAW_PATH = ".gradio/images/raw"
+EMBEDDINGS_PATH = ".gradio/images/embeddings"
+
+# Function to display images based on user selection
+def display_images(percentage, complexity):
+    # Format the filenames based on user selections
+    raw_image_path = f"{RAW_PATH}/percentage_{percentage}_complexity_{complexity}.png"
+    embeddings_image_path = f"{EMBEDDINGS_PATH}/percentage_{percentage}_complexity_{complexity}.png"
+    
+    # Return the corresponding images for raw and embeddings
+    return raw_image_path, embeddings_image_path
+
+# Define the Gradio interface
+# Step 2: Pass arrays of values for data percentage and task 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"),  # Dropdown for data percentage
+        gr.Radio(task_complexity_options, label="Task Complexity")  # Radio for 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."
+)
+
+if __name__ == "__main__":
+    demo.launch()