Update app.py

app.py

@@ -13,7 +13,7 @@ def load_model(file_path):
     model = project.version(3).model
 
     # Load the model weights into a PyTorch model
-    pytorch_model = YOLO('yolov8n.yaml')
+    pytorch_model = YOLO('args.yaml')
     pytorch_model.load_state_dict(torch.load(file_path, map_location=torch.device('cpu')))
     pytorch_model.eval()