Update app.py

app.py

@@ -5,7 +5,7 @@ from ultralytics import YOLO
 import matplotlib.pyplot as plt
 import io
 from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
-model = YOLO('detect-best.pt')
+model = YOLO('detect-best.pt', weights_only=True)
 
 def predict(img, conf, iou):
     results = model.predict(img, conf=conf, iou=iou)