Update app.py

app.py

@@ -11,7 +11,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model_file = 'model.pth'
 
 if torch.cuda.is_available():
-    model = torch.load(model_file, map_location=torch.device(torch_device))
+    model = torch.load(model_file, map_location=torch.device(device))
     model = model.eval()
 
 transform = transforms.Compose([