Spaces:
Sleeping
Sleeping
AdilzhanB
commited on
Commit
·
9c08496
1
Parent(s):
80af471
App change
Browse files
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([
|
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})
|