Update app.py

app.py

@@ -1,6 +1,6 @@
 import torch
 from torch import nn
-from transformers import AutoImageProcessor, AutoModel
+from transformers import AutoImageProcessor, Dinov2Model
 from PIL import Image
 import base64
 from io import BytesIO
@@ -12,7 +12,7 @@ MODEL_NAME = "facebook/dinov2-small"
 
 # Carregando processador e modelo
 processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
-model = AutoModel.from_pretrained(MODEL_NAME)
+model = Dinov2Model.from_pretrained(MODEL_NAME)
 
 # Projeção para 512D
 projection = nn.Linear(model.config.hidden_size, 512)