Balajim57 commited on
Commit
fca893e
·
verified ·
1 Parent(s): bf57be6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'X:\mowito\mowito.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"])
 
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"])