Update code.py

code.py

@@ -1,7 +1,7 @@
 import torch
 from yolov5 import YOLOv5
 from art.attacks.evasion import FastGradientMethod
-from art.classifiers import PyTorchClassifier
+from art.estimators.classification import PyTorchClassifier  # Alteração para nova versão do ART
 from PIL import Image
 import numpy as np
 import matplotlib.pyplot as plt
@@ -28,6 +28,7 @@ def predict_fn(x):
     with torch.no_grad():
         return model(x).xywh[0][:, 4].cpu().numpy()  # Retorna as probabilidades de deteção de objetos
 
+# Criando o classificador ART
 classifier = PyTorchClassifier(
     model=model.model,  # Passando o modelo YOLO para ART
     loss_fn=torch.nn.CrossEntropyLoss(),