Made Changes

app.py

@@ -19,16 +19,16 @@ args = DotMap()
 args.deploy = 'vanilla'
 args.arch = 'dino_small_patch16'
 args.no_pretrain = True
-args.resume = 'https://huggingface.co/ArtificialCoder02/Eurosat_Meta_3Epoch/blob/main/best.pth'
-args.api_key = 'AIzaSyAFkOGnXhy-2ZB0imDvNNqf2rHb98vR_qY'
-args.cx = '06d75168141bc47f1'
+args.resume = 'https://huggingface.co/ArtificialCoder02/Eurosat_Meta_3Epoch/blob/main/checkpoint.pth'
+args.api_key = 'hAIzaSyC5ik1DoDzUNonRnrAi3T4xP8QYtcwvHS8'
+args.cx = '772f0ccc480364116'
 
 
 # model
 device = 'cpu' #torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = get_model(args)
 model.to(device)
-checkpoint = torch.hub.load_state_dict_from_url(args.resume, map_location='cpu')
+checkpoint = torch.hub.load_state_dict_from_url(args.resume, map_location='cuda')
 model.load_state_dict(checkpoint['model'], strict=True)