Update app.py

app.py

@@ -23,7 +23,7 @@ session = ort.InferenceSession(model_path)
 # Define the inference function
 def inference(sample_name):
     sample_path = os.path.join(os.getcwd(),'samples', f'{sample_name}.pth')
-    sample = torch.load(sample_path)
+    sample = torch.load(sample_path, weights_only=False, map_location="cpu")
     age = sample['boneage'].item()
     outputs = session.run(None, {"input": sample['path'].numpy()})
     predicted_age = (outputs[0]*41.172)+127.329