Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ session = ort.InferenceSession(model_path)
|
|
17 |
|
18 |
# Define the inference function
|
19 |
def inference(sample_name):
|
20 |
-
sample_path = os.path.join(os.getcwd(),f'{sample_name}.pth')
|
21 |
sample = torch.load(sample_path)
|
22 |
age = sample['boneage'].item()
|
23 |
outputs = session.run(None, {"input": sample['path'].numpy()})
|
|
|
17 |
|
18 |
# Define the inference function
|
19 |
def inference(sample_name):
|
20 |
+
sample_path = os.path.join(os.getcwd(),'samples', f'{sample_name}.pth')
|
21 |
sample = torch.load(sample_path)
|
22 |
age = sample['boneage'].item()
|
23 |
outputs = session.run(None, {"input": sample['path'].numpy()})
|