Update app.py

app.py

@@ -8,6 +8,7 @@ from torchvision.ops import nms, box_iou
 import torch.nn.functional as F
 from torchvision import transforms
 from PIL import Image, ImageDraw, ImageFont, ImageFilter
+from sklearn.cluster import KMeans
 from data_manager import get_dog_description
 from urllib.parse import quote
 from ultralytics import YOLO