Spaces:
Sleeping
Sleeping
File size: 2,510 Bytes
959e306 3615160 32661f6 3615160 a2d322d 9a9bdf3 3615160 959e306 c387c55 dc3055d 3615160 c387c55 3615160 c387c55 dc3055d 3615160 c387c55 ded489d c387c55 3615160 959e306 a2d322d fe8c8cb a2d322d 3615160 959e306 a2d322d d4a767a fe8c8cb a2d322d 959e306 3615160 |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
import gradio as gr
import os
# Paths for images
yolov5_result = os.path.join(os.getcwd(), "data/xai/yolov5.png")
yolov8_result = os.path.join(os.getcwd(), "data/xai/yolov8.png")
yolov5_dff = os.path.join(os.getcwd(), "data/xai/yolov5_dff.png")
yolov8_dff = os.path.join(os.getcwd(), "data/xai/yolov8_dff.png")
# Netron HTML templates
def get_netron_html(model_url):
return f"""
<div style="background-color: black; padding: 1px; border: 0.5px solid white;">
<iframe
src="{model_url}"
width="100%"
height="800"
frameborder="0">
</iframe>
</div>
"""
# URLs for Netron visualizations
yolov5_url = "https://netron.app/?url=https://huggingface.co/FFusion/FFusionXL-BASE/blob/main/vae_encoder/model.onnx"
yolov8_url = "https://netron.app/?url=https://huggingface.co/spaces/BhumikaMak/NeuralVista/resolve/main/weight_files/yolov8s.pt"
custom_css = """
body {
background-color: white;
background-size: 1800px 1800px;
height: 100%;
margin: 0;
overflow-y: auto;
}
#neural-vista-title {
color: #800000 !important;
font-size: 32px;
font-weight: bold;
text-align: center;
}
#neural-vista-text {
color: #800000 !important;
font-size: 18px;
font-weight: bold;
text-align: center;
}
#highlighted-text {
font-weight: bold;
color: #1976d2;
}
"""
with gr.Blocks(css=custom_css) as demo:
gr.HTML("""
<div style="border: 2px solid #a05252; padding: 20px; border-radius: 8px;">
<span style="color: #800000; font-family: 'Papyrus', cursive; font-weight: bold; font-size: 32px;">NeuralVista</span><br><br>
<span style="color: black; font-family: 'Papyrus', cursive; font-size: 18px;">A harmonious framework of tools <span style="color: red; font-family: 'Papyrus', cursive; font-size: 18px;">☼</span> designed to illuminate the inner workings of AI.</span>
</div>
""")
with gr.Row():
with gr.Column():
gr.HTML(get_netron_html(yolov5_url))
gr.Image(yolov5_result, label="Detections & Interpretability Map")
gr.Image(yolov5_dff, label="Feature Factorization & discovered concept")
with gr.Column():
gr.HTML(get_netron_html(yolov8_url))
gr.Image(yolov8_result, label="Detections & Interpretability Map")
gr.Image(yolov8_dff, label="Feature Factorization & discovered concept")
demo.launch()
|