Update app.py

app.py

@@ -8,7 +8,7 @@ device = torch.device('cuda' if torch.cuda.is_available()  else 'cpu')
 
 os.system("wget https://www.dropbox.com/s/grcragozd4x79zc/model_best.pth?dl=0")
 
-model = torch.load("./model_best.pth?dl=0", map_location=device)
+model = torch.load("./Desktop/model_best.pth?dl=0", map_location=device)
 
 # img  = Image.open(path).convert('RGB')
 from torchvision import transforms