Spaces:
Runtime error
Runtime error
NadiaMostafa
commited on
Commit
•
701a3c8
1
Parent(s):
889e69b
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import numpy as np
|
|
6 |
import tempfile
|
7 |
|
8 |
# YOLOv5 Model Loading (best.pt)
|
9 |
-
model = torch.hub.load('ultralytics/yolov5', 'custom', path='
|
10 |
|
11 |
# Streamlit UI
|
12 |
st.title('YOLOv5 Object Detection')
|
|
|
6 |
import tempfile
|
7 |
|
8 |
# YOLOv5 Model Loading (best.pt)
|
9 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt') # Replace 'best.pt' with your model's path
|
10 |
|
11 |
# Streamlit UI
|
12 |
st.title('YOLOv5 Object Detection')
|