NadiaMostafa commited on
Commit
701a3c8
1 Parent(s): 889e69b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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='go.pt') # Replace 'best.pt' with your model's 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')