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
|