Medvira commited on
Commit
3854934
·
verified ·
1 Parent(s): b567634

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()})