Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -316,7 +316,7 @@ def main():
|
|
| 316 |
|
| 317 |
# Load models
|
| 318 |
inception_model, device = load_inception_model(r'anamoly30.pth') # Ensure 'anamoly30.pth' is in the same directory
|
| 319 |
-
yolo_model = load_yolo_model(r'
|
| 320 |
|
| 321 |
# Upload the image
|
| 322 |
uploaded_file = st.file_uploader("Choose an image", type=["jpg", "jpeg", "png"])
|
|
|
|
| 316 |
|
| 317 |
# Load models
|
| 318 |
inception_model, device = load_inception_model(r'anamoly30.pth') # Ensure 'anamoly30.pth' is in the same directory
|
| 319 |
+
yolo_model = load_yolo_model(r'best.pt') # Ensure 'yolov8n.pt' is in the same directory or specify the path
|
| 320 |
|
| 321 |
# Upload the image
|
| 322 |
uploaded_file = st.file_uploader("Choose an image", type=["jpg", "jpeg", "png"])
|