Update app.py

app.py

@@ -5,7 +5,8 @@ from PIL import Image
 import torchvision.transforms as transforms
 from functools import partial
 from dinov2 import DinoVisionTransformer
-from dinov2layers import Block, MemEffAttention
+from dinov2layers.block import Block
+from dinov2layers.attention import MemEffAttention
 
 # 定义模型（使用 vit_small 作为示例）
 def create_model(height, width):