Update app.py

app.py

@@ -22,6 +22,7 @@ classify_model = models.DinoVisionTransformerClassifier(num_classes)
 classify_model.load_state_dict(torch.load("best_dinov2_both_2023-11-21_07-44-35.pth", map_location=torch.device('cpu')))
 classify_model.eval()
 
+k = 5
 
 def classify(image):
     output = classify_model(image)[0]