Update app.py

app.py

@@ -109,7 +109,7 @@ def app_interface():
             v2.Normalize(mean=[0.485, 0.456, 0.406],
                          std=[0.229, 0.224, 0.225])])
         class_map = return_classes_()
-        model_ = torch.load(f"models/{model_1}.pt", map_location=torch.device('cpu'), weights_only = True)
+        model_ = torch.load(f"models/{model_1}.pt", map_location=torch.device('cpu'))
         # img_ = Image.open(path_)
         img__ = transform(path_).unsqueeze(dim = 0)
         pred_ = model_(img__)