Hugging Face's logo

Spaces:
AmirKaseb
/
SignLanguageModel
Build error

App Files Community
AmirKaseb commited on
Commit
fb54041
·
verified ·
1 Parent(s): b98bc72

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') # 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')