AdilzhanB commited on
Commit
9c08496
·
1 Parent(s): 80af471

App change

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def predict_bone_age(image, gender):
45
  img_tensor = transform(image).unsqueeze(0).numpy()
46
  # Gender: 0=male, 1=female
47
  gender_val = 0.0 if gender.lower() == "male" else 1.0
48
- gender_tensor = np.array([[gender_val]], dtype=np.float32)
49
 
50
  # ONNX inference
51
  outputs = ort_session.run(None, {"image": img_tensor, "gender": gender_tensor})
 
45
  img_tensor = transform(image).unsqueeze(0).numpy()
46
  # Gender: 0=male, 1=female
47
  gender_val = 0.0 if gender.lower() == "male" else 1.0
48
+ gender_tensor = np.array([gender_val], dtype=np.float32)
49
 
50
  # ONNX inference
51
  outputs = ort_session.run(None, {"image": img_tensor, "gender": gender_tensor})