Spaces:
Sleeping
Sleeping
update: added yolov8 netron support
Browse files
app.py
CHANGED
@@ -48,6 +48,15 @@ def view_model(selected_models):
|
|
48 |
"""Generate Netron visualization for the selected models."""
|
49 |
netron_html = ""
|
50 |
for model in selected_models:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
if model == "yolov5":
|
52 |
netron_html = f"""
|
53 |
<iframe
|
|
|
48 |
"""Generate Netron visualization for the selected models."""
|
49 |
netron_html = ""
|
50 |
for model in selected_models:
|
51 |
+
if model=="yolov8":
|
52 |
+
netron_html = f"""
|
53 |
+
<iframe
|
54 |
+
src="https://netron.app/?url=https://huggingface.co/spaces/BhumikaMak/NeuralVista/resolve/main/weight_files/yolov8s.pt"
|
55 |
+
width="100%"
|
56 |
+
height="800"
|
57 |
+
frameborder="0">
|
58 |
+
</iframe>
|
59 |
+
"""
|
60 |
if model == "yolov5":
|
61 |
netron_html = f"""
|
62 |
<iframe
|