Upload folder using huggingface_hub

app.py

@@ -24,8 +24,7 @@ class_labels= ['Beagle', 'Boxer', 'Bulldog', 'Dachshund', 'German_Shepherd', 'Go
 
 
 # Model
-if device=='cuda': model = torch.jit.load('best_model.pt').to(device)
-else: model=torch.load('best_model.pt')
+model = torch.jit.load('best_model.pt',map_location=device).to(device)
 
 
 @torch.no_grad()