Spaces:
Running
Running
File size: 1,502 Bytes
94f5248 5608218 94f5248 5608218 94f5248 5608218 94f5248 5608218 94f5248 5608218 94f5248 5608218 94f5248 5608218 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
# -*- coding: utf-8 -*-
"""
Created on Tue Sep 17 15:03:39 2024
@author: salikha4
"""
import gradio as gr
import os
from PIL import Image
# Paths to the images folder
RAW_PATH = os.path.join("images", "raw")
EMBEDDINGS_PATH = os.path.join("images", "embeddings")
# Function to load and display images based on user selection
def display_images(percentage, complexity):
# Generate the paths to the images
raw_image_path = os.path.join(RAW_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
embeddings_image_path = os.path.join(EMBEDDINGS_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
# Load images using PIL
raw_image = Image.open(raw_image_path)
embeddings_image = Image.open(embeddings_image_path)
# Return the loaded images
return raw_image, embeddings_image
# Define the Gradio interface
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."
)
if __name__ == "__main__":
demo.launch()
|